【安装指南】pytorch相关软件安装

Anaconda安装

PyCharm安装

PyTorch安装

cuDNN的三个文件夹复制到CUDA目录下。检查是否安装成功：

# 进入cuda的安装目录下bin目录，执行：
nvcc -V

# 进入cuda的安装目录下extras/demo_suite目录，执行：
banwidthTest.exe
deviceQuery.exe

# 查看gpu运行情况：
nvidia-smi

检查安装是否成功：

import torch

print("Hello PyTorch {}".format(torch.__version__)) # 查看torch版本

print(torch.cuda.is_available()) # 查看cuda是否可用

【安装指南】pytorch相关软件安装

原文：https://www.cnblogs.com/yanqiang/p/12749030.html