1、pytorch版本安装(卸载也会有问题)
(1)版本如何选择参考和卸载
https://zhuanlan.zhihu.com/p/401931724
(2)对应版本如何安装命令
https://pytorch.org/get-started/previous-versions/
最简答安装参考如下步骤:安装cuda cudnn pytorch
安装 pytorch 默认把对应的 cuda cudnn 安装好了
(1)cudnn和cuda版本关联一起,查看匹配的版本号,例如cuda11.0对应cudnn v8.0.5
https://developer.nvidia.com/rdp/cudnn-archive
(2)pytorch 版本和对应的cuda cudnn版本安装命令 例如我想安装pytorch1.7.1,找到如下安装命令 pytorch 需要 cu11.0 ,cu11.0 又对应cudnn v8.0.5
https://pytorch.org/get-started/previous-versions/
(3) 如上安装结束后,cuda cudnn 也会被自动安装好,(pip 超时,直接把安装包下载,本地安装)
查看pytorch 版本和对应cuda cudnn 版本是不是正确
激活安装的虚拟环境后
python
import torch
print(torch.__version__)
print(torch.version.cuda)
print(torch.backends.cudnn.version())