pytorch GPU模型训练的环境搭建

1、GPU更新并查看最高支持的cuda版本

 验证cmd命令

nvidia-smi

 

 

2、Python（anaconda）

 

 

 

3、Pytorch

https://pytorch.org/get-started/locally/ 选择自己想要安装的版本

 

 

pycharm中可以自由的选择当前项目想要的环境

验证pytorch和cuda版本

import torch
print(torch.cuda.is_available())
print(torch.backends.cudnn.is_available())
print(torch.cuda_version)
print(torch.backends.cudnn.version())