Pytorch

开发环境

  • python3.7以上
  • CUDA根据自己选择版本
  • Pycharm

CUDA的安装

版本查看

image-20230927222214275

CUDA官网下载

选择好对应的版本后，下载

image-20230927222653885

安装

下载完成后，安装，一直下一步就好

期间可能会出现屏幕闪烁，这是正常的🐶

漫长的等待后……完成安装

检查是否安装成功

打开cmd，输入nvcc -V，能输出下图的版本号即安装成功

image-20230928081704062

Pycharm和conda创建虚拟环境

使用conda创建环境_速通版

Pytorch的学习