欢迎来到尧图网

客户服务 关于我们

您的位置:首页 > 健康 > 养生 > conda虚拟环境,安装pytorch cuda cudnn版本一致,最简单方式

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

2025/5/9 2:23:20 来源:https://blog.csdn.net/m0_37192554/article/details/128631693  浏览:    关键词:conda虚拟环境,安装pytorch cuda cudnn版本一致,最简单方式

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

版权声明:

本网仅为发布的内容提供存储空间,不对发表、转载的内容提供任何形式的保证。凡本网注明“来源:XXX网络”的作品,均转载自其它媒体,著作权归作者所有,商业转载请联系作者获得授权,非商业转载请注明出处。

我们尊重并感谢每一位作者,均已注明文章来源和作者。如因作品内容、版权或其它问题,请及时与我们联系,联系邮箱:809451989@qq.com,投稿邮箱:809451989@qq.com

热搜词