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

在服务器上使用conda创建的Python 2.7环境中,通过conda安装的PyTorch和torchvision在将模型迁移到GPU时遇到延迟问题。解决方法包括卸载并使用pip重新安装PyTorch,或指定清华大学镜像源进行pip安装。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

问题描述:在服务器上用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 
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值