最新pytorch-CUDA版本pip安装

目录

1.查看显卡支持CUDA版本号

2.pytorch查询torch-对应CUDA版本并下载

 3.下面我们来给所需环境安装pytorch

4.测试安装

---

1.查看显卡支持CUDA版本号

打开控制面板找到并打开NVIDIA控制面板

点击左下角系统信息

点击组件第三行就可以看到我的可支持11.6

 

 

2.pytorch查询torch-对应CUDA版本并下载

torch对应CUDA版本

进入CUDA官网下载对应版本的CUDA不要关待会还要用到

这里我安装的是10.2版本

 

 选择第一个下载（下载基础的就可以了，下面两个是补丁后续需要可以自取）

下载成功后打开exe文件选择环境安装路径（默认C盘）

 3.所需环境安装pytorch

我用的是pycharm，打开项目文件目录

 进入Scipts文件然后在上面索引框输入cmd进入到该目录

然后打开我们打开pytorch对应CUDA版本找到要下载的版本

 

 先安装whl输入OSX（安装）下面的单行安装代码

然后输入下面自己的CUDA版本安装代码我是10.2

这里我已经安装过了，未安装的安装成功会显示successful

4.测试安装

打开pycharm新建一个test.py文件

输入图中代码即可测试

import torch

x = torch.rand(5, 3)
print(x)

print(torch.cuda.is_available())

 安装成功！

大家有问题可以提，第一次发有做的不好的地方希望大家批评指正！