tips:
1)注意pip命令安装与conda命令安装,torch对应的cuda版本可能不一样,具体见官网
2)cuda10.2,cuda11.3可以对应多个版本的torch,比较稳定。其中cuda10.2搭配cuDNN7.6.5最稳定可靠
1.查对应版本
(1)查版本
TensorFlow、PyTorch各版本对应的CUDA、cuDNN关系_C_GH的博客-CSDN博客
torch版本最高到1.8.0
Pytorch环境配置——cuda、、cudnn、torch、torchvision对应版本(最全)及安装方法_cuda12.2对应的pytorch_William.csj的博客-CSDN博客
最新,torch版本到1.10.1
PyTorch和CUDA版本对应关系 - 墨莲玦 - 博客园 (cnblogs.com)
01 Pytorch和CUDA对应的版本及Pytorch和Python对应的版本及Python与Anaconda的对应关系_python和cuda版本-CSDN博客
pytorch与python对应版本
(2)最好用ÿ