首页 最新 热门 推荐

  • 首页
  • 最新
  • 热门
  • 推荐

Ubuntu下使用conda在虚拟环境中安装CUDA、CUDNN及Tensorflow

  • 25-02-18 05:20
  • 4091
  • 13484
blog.csdn.net

在Ubuntu中安装Anaconda可以根据 这篇博客,写的十分详细。
装好之后,可以通过通过命令安装版本为X.X(如2.7,3.6)的虚拟环境:

conda create -n your_env_name python=X.X
  • 1

其中(your_env_name)是你所创建的虚拟环境的名字。
然后 使用下方命令激活你刚才建立的环境

source activate your_env_name
  • 1

激活了环境之后,我们可以看到命令行之前多了一个括号,括号中包含了你的环境名字,这就说明激活成功。
在激活的环境中使用如下命令安装CUDA:

conda install cudatoolkit=8.0 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/linux-64/
  • 1

我装CUDA的同时,自动装上了CUDNN。
如果你没有装上,也可以使用下方命令安装CUDNN:

conda install cudnn=7.0.5 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/linux-64/
  • 1

其中CUDA的版本可以自己更改,我改成了9.2.0,CUDNN应该也可以改,不过我没有尝试。
这篇文章还展示了Tensorflow不同版本要求与CUDA及CUDNN版本对应关系,一般我们都是装好了tensorflow-gpu,然后需要对应版本的CUDA及CUDNN,可以让大家参考一下。
仍然是在激活的环境中安装tensorflow,比如我的CUDA是9.2.0,我的CUDNN是7.2.1,那么对应的Tensorflow就可以安装1.5.0,如下方命令所示:

pip install tensorflow-gpu==1.5.0
  • 1

以上所安装的Python、CUDA、CUDNN以及Tensorflow都是在虚拟环境中安装的,下次想要使用该环境时或者在该环境安装其他库包文件,需要重新进入虚拟环境。与之前的代码一样,如下:

source activate your_env_name
  • 1

查询当前环境下的库的版本号可使用以下命令:

conda list cudnn
conda list cuda
  • 1
  • 2
注:本文转载自blog.csdn.net的小小玺先生的文章"https://blog.csdn.net/weixin_40588315/article/details/85881338"。版权归原作者所有,此博客不拥有其著作权,亦不承担相应法律责任。如有侵权,请联系我们删除。
复制链接
复制链接
相关推荐
发表评论
登录后才能发表评论和回复 注册

/ 登录

评论记录:

未查询到任何数据!
回复评论:

分类栏目

后端 (14832) 前端 (14280) 移动开发 (3760) 编程语言 (3851) Java (3904) Python (3298) 人工智能 (10119) AIGC (2810) 大数据 (3499) 数据库 (3945) 数据结构与算法 (3757) 音视频 (2669) 云原生 (3145) 云平台 (2965) 前沿技术 (2993) 开源 (2160) 小程序 (2860) 运维 (2533) 服务器 (2698) 操作系统 (2325) 硬件开发 (2492) 嵌入式 (2955) 微软技术 (2769) 软件工程 (2056) 测试 (2865) 网络空间安全 (2948) 网络与通信 (2797) 用户体验设计 (2592) 学习和成长 (2593) 搜索 (2744) 开发工具 (7108) 游戏 (2829) HarmonyOS (2935) 区块链 (2782) 数学 (3112) 3C硬件 (2759) 资讯 (2909) Android (4709) iOS (1850) 代码人生 (3043) 阅读 (2841)

热门文章

101
推荐
关于我们 隐私政策 免责声明 联系我们
Copyright © 2020-2025 蚁人论坛 (iYenn.com) All Rights Reserved.
Scroll to Top