下载好anaconda和pychram,可以直接在pycharm终端配置深度学习环境的配置。
一、查看可支持cuda版本
使用nvidia-smi命令在cmd窗口,查看显卡最高支持的cuda版本,如下:
表示安装的cuda版本不能超过11.2,具体版本视自己电脑而定。
二、配置环境
在安装torch时候,需要注意的是版本之间的对应,需要版本一致,其官网的对应关系如下链接。
PyTorch中torch、torchvision、torchaudio版本对应关系_torch与torchvision版本_Liekkas Kono的博客-CSDN博客