pytorch GPU模型训练的环境搭建

大背头 / 2023-07-28 / 原文

1、GPU(CUDA、cuDNN)

 验证cmd命令

nvidia-smi

ncvv -V

 

 

2、Python(anaconda)

 

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

 

3、Pytorch