安装pytorch

1、下载英伟达的cuda与cudnn（对应电脑版本）

如何查询电脑可以用什么版本 打开cmd 输入nvidia-smi (信息中可以看到)

紧接着去英伟达官网下载对应的cuda 以及cudnn

2、将解压的cudnn文件内容复制在安装的cuda中

3、打开anaconda prompt 下载包

创建pytorch包 conda create -n pytorch python=3.8 根据自己的版本决定

4、conda activate pytorch进入pytorch

 

 5、pytorch官网下载

 在命令行中输入官网的命令下载

使用conda list查看下载的东西

6、测试是否安装成功

1.在命令行输入python

2.输入import torch

3.在输入torch.__version__

4.torch.cuda.is_available()

如下图则算成功

 7、连接pycharm测试

 

测试用例

import torch

print("hello {}".format(torch.__version__))
print(torch.cuda.is_available())

 如下图结果则表示成功

 恭喜安装pytorch 完结撒花