找回密码
 会员注册
查看: 22|回复: 0

Python:condainstallcudatoolkit的备选方案:pipinstallnvidia-cudnn-:

[复制链接]

2万

主题

0

回帖

6万

积分

超级版主

积分
69864
发表于 2024-9-10 06:50:43 | 显示全部楼层 |阅读模式
由于使用深度学习框架的不同,有的时候我们需要切换cudnn环境。比起在系统中安装多个cudnn版本,更便捷的方法是通过在python环境下安装cudnn工具,这样不同的cudnn环境就可以用python的包管理器(如conda等)管理,使用起来很方便。最常用的方式是在conda下,通过安装不同版本的cudatoolkit来满足要求。condainstallcudatoolkit1然而有的时候我们用的包管理器不是cuda,或者我们用的python包镜像不支持cuda,这时只能用pip.以cuda11为例,此时可以使用以下指令安装需要的cudnn工具;注意选择自己需要的版本号。pipinstallnvidia-cudnn-cu11pipinstallnvidia-cuda-runtime-cu11pipinstallnvidia-cublas-cu11pipinstallnvidia-cuda-nvrtc-cu111234
回复

使用道具 举报

您需要登录后才可以回帖 登录 | 会员注册

本版积分规则

QQ|手机版|心飞设计-版权所有:微度网络信息技术服务中心 ( 鲁ICP备17032091号-12 )|网站地图

GMT+8, 2025-1-7 07:06 , Processed in 0.448636 second(s), 26 queries .

Powered by Discuz! X3.5

© 2001-2025 Discuz! Team.

快速回复 返回顶部 返回列表