NVIDIA控制面板-》管理3D设置-》系统信息-》组件查看对应的CUDA版本,我的是11.4,因为pytorch还没跟新到11.4版本,所以只能安装11.3版本的了。
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
测试一下