win10+Anoconda+Vscode安装python=3.9的pytorch环境

环境： win10 + Anoconda + Vscode

1. 创建虚拟环境

conda create -n pytorch39 python=3.9

2. 激活环境

conda activate pytorch39

3. 查看cuda版本。

   搜索NVIDIA CONTROL PANEL。

   

   点左下角【系统信息】。
   

   切换到组件。
   

4. 到官网Start Locally | Pytorch寻找符合版本的安装命令，并执行。我的cuda版本是11.6所以选11.6的安装。

conda install pytorch torchvision torchaudio pytorch-cuda=11.6 -c pytorch -c nvidia

5. 检查torch安装好了没，要是True且有版本号。先python进入python环境，再输入接下来的代码。

python

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