1. CUDA11.6
2. Anaconda4.10.3
3. 安装torch,在Anaconda prompt中依次输入以下命令:
conda config --set show_channel_urls yes
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
4. 验证是否安装成功
pip show torch
import torch
torch.cuda.is_available()
综上
CUDA 版本:11.6
Anaconda版本:4.10.3
pytorch版本:1.10.2+cu113