广告
返回顶部
首页 > 资讯 > 后端开发 > Python >Python:conda install cudatoolkit的备选方案:pip install nvidia-cudnn-*:
  • 697
分享到

Python:conda install cudatoolkit的备选方案:pip install nvidia-cudnn-*:

pythonpipcondacudacudnn 2023-09-01 08:09:46 697人浏览 八月长安

Python 官方文档:入门教程 => 点击学习

摘要

由于使用深度学习框架的不同,有的时候我们需要切换cudnn环境。比起在系统中安装多个cudnn版本,更便捷的方法是通过在python环境下安装cudnn工具,这样不同的cudnn环境就可以用Pytho

由于使用深度学习框架的不同,有的时候我们需要切换cudnn环境。比起在系统中安装多个cudnn版本,更便捷的方法是通过在python环境下安装cudnn工具,这样不同的cudnn环境就可以用Python的包管理器(如conda等)管理,使用起来很方便。
最常用的方式是在conda下,通过安装不同版本的cudatoolkit来满足要求。

conda install cudatoolkit

然而有的时候我们用的包管理器不是cuda,或者我们用的python包镜像不支持cuda,这时只能用pip. 以cuda11为例,此时可以使用以下指令安装需要的cudnn工具;注意选择自己需要的版本号。

pip install nvidia-cudnn-cu11pip install nvidia-cuda-runtime-cu11pip install nvidia-cublas-cu11pip install nvidia-cuda-nvrtc-cu11

来源地址:https://blog.csdn.net/WinterShiver/article/details/131923448

--结束END--

本文标题: Python:conda install cudatoolkit的备选方案:pip install nvidia-cudnn-*:

本文链接: https://www.lsjlt.com/news/387827.html(转载时请注明来源链接)

有问题或投稿请发送至: 邮箱/279061341@qq.com    QQ/279061341

本篇文章演示代码以及资料文档资料下载

下载Word文档到电脑,方便收藏和打印~

下载Word文档
猜你喜欢
软考高级职称资格查询
编程网,编程工程师的家园,是目前国内优秀的开源技术社区之一,形成了由开源软件库、代码分享、资讯、协作翻译、讨论区和博客等几大频道内容,为IT开发者提供了一个发现、使用、并交流开源技术的平台。
  • 官方手机版

  • 微信公众号

  • 商务合作