Win11下Pytorch1.12.0+cuda11.6环境配置

注：我的笔记本显卡是GeForce RTX4060

pytorch1.12.0，适合的python版本3.7、3.8、3.9、3.10，这里选择3.9的版本

创建conda虚拟环境并命名为pytorch1.12，命令如下：

conda create -n pytorch1.12 python==3.9

激活当前虚拟环境

activate pytorch1.12

根据Previous PyTorch Versions | PyTorch ，安装命令如下：

# CUDA 11.6
conda install pytorch==1.12.0 torchvision==0.13.0 torchaudio==0.12.0 cudatoolkit=11.6 -c pytorch -c conda-forge
# CPU Only
conda install pytorch==1.12.0 torchvision==0.13.0 torchaudio==0.12.0 cpuonly -c pytorch

 安装完成后输入python，输入下面的命令查看pytorch版本是否安装正确

import torch # 如果pytorch安装成功即可导入
print(torch.cuda.is_available()) # 查看CUDA是否可用
print(torch.cuda.device_count()) # 查看可用的CUDA数量
print(torch.version.cuda) # 查看CUDA的版本号

结果如下即证明安装成功