conda安装pytotch和torchvision，加载模型到GPU中等待时间过长

本文主要是介绍conda安装pytotch和torchvision，加载模型到GPU中等待时间过长，希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！

问题描述：在服务器上用conda创建了自己的环境，然后在该环境下用conda创建一个python 2.7的环境，在该环境下安装pytorch和torchvision包，训练模型时，model = model.cuda()占据过多时间。

解决：
卸载之前安装的pytorch，conda uninstall pytorch
然后pip重新安装，pip install xx.whl # 把whl下载到本地安装

或者，直接输入链接安装

pip install http://download.pytorch.org/whl/cu80/torch-0.3.0.post4-cp27-cp27mu-linux_x86_64.whl

pip安装包时-i 参数选择源，比如说，
清华源：https://pypi.tuna.tsinghua.edu.cn/simple/

**可以先去清华源网站上搜索有没有你想要的python包**
pip install --upgrade tensorflow -i https://pypi.tuna.tsinghua.edu.cn/simple
或者安装pretrainedmodels
pip install pretrainedmodels -i https://pypi.tuna.tsinghua.edu.cn/simple/

这篇关于conda安装pytotch和torchvision，加载模型到GPU中等待时间过长的文章就介绍到这儿，希望我们推荐的文章对编程师们有所帮助！

---