Ubuntu 系统上使用 .whl 文件安装 PyTorch、Torchvision

运行github上的代码,经常人家的环境是比较老的cuda和torch,这时候使用新版本会导致其余依赖库的安装失败,这篇记录一下如何安装老版本的torch(重新搞cuda太麻烦了,后续可以使用docker镜像,但我暂时不会):

去这个地方找需要的cuda版本对应的torchwhl文件:

https://download.pytorch.org/whl/torch_stable.html

一定要看好了下载,不要下载成cpu。如图是下载cuda为11.0,torch为1.7,python为3.6在linux系统下的文件

ubuntu系统基本是会给放到downloads文件夹下面,之后激活虚拟环境再下载即可,使用如下命令,把后面的文件名换了就行:

pip install ~/downloads/torch-1.7.0+cu110-cp37-cp37m-linux_x86_64.whl
pip install ~/downloads/torchcsprn-0.2.0+cu110-cp37-cp37m-linux_x86_64.whl
pip install ~/downloads/torchvision-0.8.0-cp37-cp37m-linux_x86_64.whl
 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值