Anaconda的python虚拟环境中安装cudatoolkit和cudnn加速tensorflow

这篇具有很好参考价值的文章主要介绍了Anaconda的python虚拟环境中安装cudatoolkit和cudnn加速tensorflow。希望对大家有所帮助。如果存在错误或未考虑完全的地方,请大家不吝赐教,您也可以点击"举报违法"按钮提交疑问。

1. 背景

由于本地安装了cuda 10.0, 但是现在需要在Anaconda中安装不同的python虚拟环境来安装tensorflow-gpu、对应的cudatoolkit、对应的cudnn来加速,下面是具体的演示流程

2. 安装

我这里以安装tensorflow-gpu==1.9.0为例,首先进入python的虚拟环境,source activate py37_tf

Anaconda的python虚拟环境中安装cudatoolkit和cudnn加速tensorflow,tensorflow

通过"conda search tensorflow-gpu", 可以看到有满足安装版本要求的tensorflow-gpu

Anaconda的python虚拟环境中安装cudatoolkit和cudnn加速tensorflow,tensorflow

然后使用清华源安装: pip install tensorflow-gpu==1.9.0 -i Simple Index

Anaconda的python虚拟环境中安装cudatoolkit和cudnn加速tensorflow,tensorflow

 安装成功以后去网络上搜索"tensorflow和GPU的对应关系"

Anaconda的python虚拟环境中安装cudatoolkit和cudnn加速tensorflow,tensorflow

可以看到需要对应安装cudatoolkit==9.0, 我们在通过命令: "conda search cudatoolkit"来看一下是否有此cudatoolkit的安装包,输入"conda search cudatoolkit"

Anaconda的python虚拟环境中安装cudatoolkit和cudnn加速tensorflow,tensorflow

发现是有这个cudatoolkit 9.0的,于是执行: conda install cudatoolkit==9.0

Anaconda的python虚拟环境中安装cudatoolkit和cudnn加速tensorflow,tensorflow

安装成功cudatoolkit之后,需要再安装对应版本的cudnn, 于是通过conda search cudnn查看一下,发现也是有可以安装的版本

Anaconda的python虚拟环境中安装cudatoolkit和cudnn加速tensorflow,tensorflow

于是执行: conda install cudnn==7.3.1

Anaconda的python虚拟环境中安装cudatoolkit和cudnn加速tensorflow,tensorflow

 则可以安装成功

上面的tensorflow-gpu、对应的cudatoolkit、对应的cudnn都安装好了之后,需要查看先tensorflow是否支持gpu加速,执行:

import tensorflow as tf

tf.test.is_gpu_available()

Anaconda的python虚拟环境中安装cudatoolkit和cudnn加速tensorflow,tensorflow

发现输出结果为True, 则到此支持GPU加速的tensorflow-gpu安装完毕。

参考链接:

【深度学习】不同虚拟环境安装不同Pytorch版本_不同环境安装不同pyorch_视觉闫小亘的博客-CSDN博客文章来源地址https://www.toymoban.com/news/detail-578048.html

到了这里,关于Anaconda的python虚拟环境中安装cudatoolkit和cudnn加速tensorflow的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处: 如若内容造成侵权/违法违规/事实不符,请点击违法举报进行投诉反馈,一经查实,立即删除!

领支付宝红包 赞助服务器费用

相关文章

  • Ubuntu搭建Pytorch环境(Anaconda、Cuda、cuDNN、Pytorch、Python、Pycharm、Jupyter)

    1.查看Ubuntu版本号: cat /etc/issue ,后续根据版本号添加对应的镜像源 2.备份镜像源: sudo cp /etc/apt/sources.list /etc/apt/sources.list.bak 3.在阿里云镜像源官网中找到并复制与版本号对应的阿里镜像源: 或者在清华镜像源官网中找到并复制与版本号对应的清华镜像源: 4.编辑软件源配

    2024年02月09日
    浏览(282)
  • Python Anaconda创建虚拟环境及Pycharm使用虚拟环境

    前言 一、Anaconda与Pycharm 二、conda常用命令 三、Pycharm使用虚拟环境 总结         我们在做开发任务时可能会创建多个项目,这些项目可能会依赖于不同的Python环境。比如有的用到Python3.6、有的用到Python3.7;有的用Pytorch开发、有的用TensorFlow开发。这时我们需要为不同的项

    2024年02月10日
    浏览(84)
  • Python中安装和使用venv虚拟环境

    要在Python中安装和使用venv虚拟环境,您可以按照以下步骤进行操作: 1.首先,确保您已安装了Python。您可以在命令行中输入以下命令来验证Python的安装情况: 2.在您的项目文件夹中创建一个新的虚拟环境。在命令行中,切换到您要创建虚拟环境的目录,并运行以下命令: 这

    2024年02月11日
    浏览(43)
  • 使用anaconda创建python虚拟环境

    强烈推荐下载anaconda,它可以很好地管理我们的python的包,对于pycharm使用也更灵活! 首先打开anaconda prompt命令行。 日常非常有用的conda指令 1、检查更新conda 2、查询已安装的包 3、下载安装包 4、创建python虚拟环境 如果按回车后显示的最前面的括号内是env_name便代表创建成功

    2024年02月16日
    浏览(45)
  • anaconda虚拟环境改变python版本

    比如我原来是python3.9,现在想改成3.5,有方案说直接在虚拟环境中执行: 这样是不可行的!!!因为这样只是改变了python的版本,其它包没有被改变。 但其它包当初是依赖于python3.9版本的,现在python突然变成3.5了,那依赖关系就乱了,其它包也得重装。比如包A原本是1.10版的

    2024年02月15日
    浏览(56)
  • Linux离线状态下安装cuda、cudnn、cudatoolkit

    工具包下载地址 CUDA历史版本下载地址:https://developer.nvidia.com/cuda-toolkit-archive cuDNN历史版本下载地址:https://developer.nvidia.com/rdp/cudnn-archive cudatoolkit各版本下载地址:https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/linux-64/ 版本要求 pytorch、cuda、cuDNN三者严格对应,此处安装cuda_10.2

    2024年03月23日
    浏览(33)
  • 【Python小技巧】通过Anaconda虚拟环境解决Python新版本无法运行指定脚本的问题,并实现Notepad++一键运行 Python Anaconda虚拟环境

    笔者今天碰到一个奇怪的问题,Python从3.8升级到3.10 后,一直以来还不错。但今天运行某个函数总是没反应就结束了。刚开始以为是pandas1.5.3升级到2.0.0版本的问题,当时因为2.0pandas运行很快就升级了,先排除这个。把pandas先升级到最新版,不行。后又降到1.5.3版本还是不行。

    2024年02月04日
    浏览(67)
  • 创建Anaconda虚拟Python环境的方法

      本文介绍在 Anaconda 环境下,创建、使用与删除 Python 虚拟环境的方法。   在 Python 的使用过程中,我们常常由于不同 Python 版本以及不同 第三方库 版本的支持情况与相互之间的冲突情况,而需要创建不同的 Python 虚拟环境;在 Anaconda 的帮助下,这一步骤就变得十分方便

    2024年02月09日
    浏览(51)
  • Notepad++ 配置python虚拟环境(Anaconda)

    打开Notepad++ -”运行”菜单-”运行”按钮 在弹出的窗口内输入以下命令: 我的conda中存在虚拟环境 (1) base (2) pytorch_gpu

    2024年02月17日
    浏览(69)
  • Linux下使用anaconda创建python虚拟环境

    1、创建之前确保已经安装了anaconda。 2、在linux系统终端(也可以在pycharm中的终端下)输入conda指令,进入conda环境。    3、使用指令conda create --name 环境名称 python==3.6 创建自己想要创建的环境(环境名称自己命名)。  4、显示如下,输入y继续。  5、输入conda env list 显示虚拟

    2024年02月08日
    浏览(46)

觉得文章有用就打赏一下文章作者

支付宝扫一扫打赏

博客赞助

微信扫一扫打赏

请作者喝杯咖啡吧~博客赞助

支付宝扫一扫领取红包,优惠每天领

二维码1

领取红包

二维码2

领红包