记录:之前踩了很多坑,每次装完之后一测试就是false,有点头秃。想了下,这次成功装上就来写个帖子pick一下,以后不用再一个一个的试别人的经验帖了。
首先,我们需要进入pytorch 的官方网站,百度搜索pytorch,进入官网后选择get started栏目,然后选择previous pytorch version栏目,如下图:
往下滑鼠标,选择你想要装的torch版本和torchvision版本,注意这里有两个注意的地方:
1.确定你是想装cuda版本的torch还是想装cpu版本的torch,选择正确的安装命令行复制;
2.这点我踩坑了,可以发现命令行还分两种,一种是conda安装版本,一种是wheel安装版本,反正我用conda版本安装完,发现import torch命令可以正常通过,但打印torch版本和cuda版本时就会报错,所以建议使用wheel版本的命令行进行安装比较省事;
3.安装可能会比较慢,建议使用国内镜像,如清华源,直接更改源的命令如下:
pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple
4.验证是否安装成功的测试代码如下:
import torch
print(torch.__version__)
print(torch.cuda.is_available())
打印结果如图。解释一下,cu代表torch时cuda版本:
文章出处登录后可见!
已经登录?立即刷新