在anaconda中安装pytorch

NVIDIA控制面板-》管理3D设置-》系统信息-》组件查看对应的CUDA版本，我的是11.4，因为pytorch还没跟新到11.4版本，所以只能安装11.3版本的了。

 去pytorch官网

pip3 install torch==1.10.0+cu113 torchvision==0.11.1+cu113 torchaudio===0.10.0+cu113 -f https://download.pytorch.org/whl/cu113/torch_stable.html

 

测试是否安装成功，检查GPU 驱动程序和 CUDA 是否启用并可由 PyTorch 访问

torch.cuda.is_available()

我也在wsl2里下载安装了cuda

pip3 install torch==1.10.0+cu113 torchvision==0.11.1+cu113 torchaudio==0.10.0+cu113 -f https://download.pytorch.org/whl/cu113/torch_stable.html

测试一下