Ubuntu16.04使用Conda安装Tensorflow(GPU版本)
程序员文章站
2022-06-05 20:42:24
...
0.写在前面
这是第一次使用Markdown语言,也是第一次写博客,谨以此博客正式开启深度学习之路。
前两天为了安装Tensorflow各种折腾,最后终于在别人的帮助下熟悉了这一整套流程,特此记录下来以便日后查阅。
1.使用Conda新建环境
conda create -n tf python=3.6
这里新建了一个名为tf的环境,指定Python版本为3.6。
2.在新环境里升级pip
首先**新环境:
source activate tf
然后升级pip(如果pip版本过低后面可能会报错):
pip install --upgrade pip
3.从清华镜像源安装Tensorflow
pip install tensorflow-gpu==1.10.1 -i https://pypi.tuna.tsinghua.edu.cn/simple/
本人的环境为Ubuntu16.04LTS+CUDA 9.0+cuDNN 7.1.3, 所以这里指定tensorflow版本为1.10.1(更高版本的tensorflow应该也可以,但是有可能会出现问题)。关于CUDA、cuDNN和tensorflow的版本匹配问题,参考博客:
4.安装Spyder
tensorflow的安装到上一步就算是完成了。但是由于我们是在虚拟环境里安装的,所以直接在spyder编译器里面是无法导入的。因此需要在这个环境里再安装一个spyder。
conda install spyder
对于vscode来说,可以直接在底栏选择编译器的环境,因此可以省去这一步。
5.后悔药
如果安装失败,或者想重新安装的话,可以删除掉这个环境从头再来:
conda env remove -n tf
现在就可以愉快的使用tensorflow了~
推荐阅读
-
记录:Ubuntu 18.04 安装 tensorflow-gpu 版本
-
Ubuntu16.04下安装Cuda8.0+Caffe+TensorFlow-gpu+Pycharm过程(2018年1月23)
-
华为matebook win+deepin 15.11双系统 + 双显卡 + 使用/卸载大黄蜂方案 + NVIDIA独显驱动安装 + tensorflow-gpu-2.0
-
【环境搭建】win10安装TensorFlow全过程(python+anaconda+pycharm+CPU版本+GPU版本)
-
Tensorflow2.0.0 GPU版本避坑安装+全套资源链接
-
Windows安装tensorflow GPU版本--tensorflow-gpu版本与CUDA版本对应关系
-
Ubuntu 18.04 安装 tensorflow(GPU版本)
-
2020.07月 Tensorflow-gpu 2.1版本 win10+NVIDIA GeForce MX150安装趟坑血泪安装记录
-
Ubuntu16.04使用Conda安装Tensorflow(GPU版本)
-
Ubuntu16.04安装tensorflow-GPU