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/

猜你喜欢

转载自blog.csdn.net/tsq292978891/article/details/80212250