conda虚拟环境，安装pytorch cuda cudnn版本一致，最简单方式

贝猫说python

已于 2024-06-12 10:43:45 修改

阅读量401

点赞数 5

文章标签： conda pytorch 人工智能

于 2024-06-12 10:43:01 首次发布

本文链接：https://blog.csdn.net/m0_37192554/article/details/128631693

版权

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())