当前位置:   article > 正文

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

卸载torch

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

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

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

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

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

 注意一下红框部分

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

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

3.离线安装

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

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

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/小舞很执着/article/detail/975500
推荐阅读
相关标签
  

闽ICP备14008679号