查看CUDA版本

执行nvidia-smi命令，如下图所示

image-20230726115622592

可以发现我的CUDA Version为12.0，因此可以安装cuda12.0以下的pytorch

创建虚拟环境

这里我取虚拟环境名为DL（DeepLearning），可以自定义

1
conda create -n DL python=3.9  # 创建虚拟环境并指定python版本

激活虚拟环境

1
conda activate DL

查看Anaconda库中的pytorch

1
conda search pytorch
image-20230726121236260

conda安装pytorch

找到对应的pytorch（主要看python版本、CUDA版本），使用conda进行安装 这里我指定pytorch版本为1.12.0，build号为py3.9_cuda11.6_cudnn8_0，没有指定channel

1
conda install pytorch=1.12.0=py3.9_cuda11.6_cudnn8_0

如果使用conda安装不上可以使用pip安装

百度网盘下载【提取码8mtp】