卸载cpu版本的torch并离线安装对应的gpu版本

文章介绍了如何检查和管理torch的版本,特别是在GPU和CPU版本间切换的问题。通过使用特定的命令查看torch及cuda的版本,确定是否需要卸载重装。文章推荐了离线安装方法,包括从官方资源网站下载wheel文件,然后使用pip进行安装,确保版本匹配。此方法同样适用于其他库的离线安装。
摘要由CSDN通过智能技术生成

        每次从github上安装项目对应的库,利用requirements.txt安装很容易出现版本不对应的情况,尤其是将torch的gpu版本安装成cpu。这里记录一些查看版本的指令和离线安装的方法,就不用每次百度啦!(注:其他库的离线安装也可以用同样的方法,只需要去相应的网站下载wheel即可)

1.首先查看torch版本(指令最好在终端一行一行地输入)

activate pytorch #进入环境
python
import torch
print(torch.__version__)#查看torch版本
print(torch.cuda.is_available())#查看cuda能用吗

如果版本是cpu和输出false,可以选择卸载并重新安装,这里用离线安装的方式最稳:

pip uninstall torch #卸载之前安装的torch
python -m pip debug --verbose #查看自己能安装的wheel版本

 注意一下红框部分

 2.然后下载离线资源 https://download.pytorch.org/whl/torch_stable.html

 第一个框cu表示gpu版本,102表示版本为10.2,第二个框表示torch,一般也同时安装torchversion,往下拉选择相同的版本就好,第三个框与上一步红框相对应。

3.离线安装

cd C:\Users\1\Desktop #进到你文件下载的地方
pip install torch-1.9.1+cu102-cp37-cp37m-win_amd64.whl #安装,install后面是你下载的文件名加上后缀.whl
pip install torchvision-0.10.0-cp37-cp37m-win_amd64.whl #安装torchversion

 4.然后用步骤一的指令检验一下欧克了:

评论 15
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值