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
2024-01-27 18:22:58
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