第一次在Linux下装GPU版本的TensorFlow,踩坑,折腾了2天、记录一下。
一、linux 下安装anaconda3
1、wget https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/Anaconda3-5.3.1-Linux-x86_64.sh
对应(python 3.7.x)
2、安装依赖:yum install -y bzip2
3、 bash Anaconda3-5.3.1-Linux-x86_64.sh
在安装过程中,基本上不断按回车或者yes默认就行了。
最后一步询问是否安装vscode,输入no,毕竟在服务器里面都是用vim编辑器的,没必要多装一个。
如果你想装Python3.6.x或者其他版本的,去 https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive下找对应的版本,把路径改一下就好了。
3、配置环境变量
安装完成后,我们需要对环境变量进行添加,方便我们启动。
修改profile文件
vim /etc/profile
在文件的末尾加上下述代码:
#Anaconda
export PATH=/root/anaconda3/bin:$PATH
# 这里需要注意修改路径,就是上面安装时候提示的安装路径
最后重新载入配置文件
输入source /etc/profile
这样,配置完毕环境变量。
最好重启终端一下
打开终端(Terminal),输入python,如果显示了正确的Python版本(我这里是3.7.0)就可以了。
二、安装指定版本的GPU 版TensorFlow
1、在使用conda时,请配置一下conda的镜像,建议清华镜像.(改源)这样会比较快
依次执行以下3条命令。
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --set show_channel_urls yes
设置完成后,安装时有几个包还是多次中断,最终解决方法:cd /root
ls -a显示所有文件包括隐藏文件,找到.condarc,打开,删除 - default,将https换成http。
2、创建虚拟环境
conda create -n xxx python=3.7
#xxx是你要创建的虚拟环境的名字,自定,随意
3、启动刚创建的虚拟环境
source activate xxx
#如果要关闭运行环境,执行
source deactivate xxx
4、安装指定版本的GPU 版TensorFlow
conda install tensorflow-gpu==1.12