windows和Linux系统如何安装指定版本的torch ,且cuda可用,一个命令行解决。

记录:之前踩了很多坑,每次装完之后一测试就是false,有点头秃。想了下,这次成功装上就来写个帖子pick一下,以后不用再一个一个的试别人的经验帖了。

首先,我们需要进入pytorch 的官方网站,百度搜索pytorch,进入官网后选择get started栏目,然后选择previous pytorch version栏目,如下图:

windows和Linux系统如何安装指定版本的torch ,且cuda可用,一个命令行解决。

往下滑鼠标,选择你想要装的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版本:

windows和Linux系统如何安装指定版本的torch ,且cuda可用,一个命令行解决。

 

文章出处登录后可见!

已经登录?立即刷新

共计人评分,平均

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

(0)
xiaoxingxing的头像xiaoxingxing管理团队
上一篇 2022年6月15日
下一篇 2022年6月15日

相关推荐