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

        每次从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.然后用步骤一的指令检验一下欧克了:

文章出处登录后可见!

已经登录?立即刷新

共计人评分,平均

到目前为止还没有投票!成为第一位评论此文章。

(0)
xiaoxingxing的头像xiaoxingxing管理团队
上一篇 2023年8月8日
下一篇 2023年8月8日

相关推荐