每次从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.然后用步骤一的指令检验一下欧克了:
文章出处登录后可见!
已经登录?立即刷新