写在前面,想在Ubuntu16.04安装TensorFlow-GPU,翻了很多前辈写的文章。
使用Anaconda来安装TensorFlow-GPU的话是不需要安装不需要单独安装cuda和cudnn。
步骤:
1、之前在电脑里安装过TensorFlow-CPU,先卸载了
2、安装 英伟达驱动 首先查自己电脑的显卡型号
安装过程参考了 https://blog.csdn.net/u014797226/article/details/79626693
3、确定安装anaconda的版本号,Python3.5,以及TensorFlow-GPU版本号
4、之前电脑安装过 anaconda3, 但是好像是支持Python3.7,所以要下载支持3.5版本的吧
安装anaconda3.5版本的时候参考了 https://blog.csdn.net/CGJustDoIT/article/details/91897362
5、安装anaconda3的时候由于之前安装过,卸载了看到有人说怕卸载不干净,所以另寻了安装的位置,安装完TensorFlow-GPU后,import时候出现错误,说木有six包,找了很久都没有解决,最后看到有一篇文章中写有自定义安装anaconda后,激活环境和安装命令不太一样,所以卸载了。
6、默认了anaconda3的安装路径
安装步骤是这个 https://blog.csdn.net/CGJustDoIT/article/details/91897362
7、安装完成后,创建虚拟环境:conda create -n tf-gpu python=3.5
tf-gpu是创建的环境的名字,可以修改
Python=3.5是对环境的限定
8、创建好conda虚拟环境后,在使用之前必须先进行激活。下面激活刚创建的tensorflow虚拟环境,命令如下:
conda source activate tf-gpu
如果要注销退出当前的虚拟环境,则执行命令:
conda source deactivate tf-gpu
9、激活虚拟环境之后会提示安装很多包,安就好了
10、在安装TensorFlow-GPU的时候查可安装的版本 :conda search tensorflow-gpu
11、我根据Python版本号选择了 1.2.1版本,命令:conda install tensorflow-gpu=1.2.1
12、安装完毕之后测试是否安装成功,测试代码
然后出现了下边的错误:
然后找了找答案说是没有事,可能是第一次,所以等会就好咯,最后成功!
写在后边:感谢文中前辈们写的文章,前前后后安装了很久,记录一下,嗯希望能帮助到其他人!