第一步 GPU驱动安装
如果测试运行nvidia-smi
,有表格输出说明GPU驱动已安装,可跳过该步骤。
-
查看linux的版本以及GPU信息
(base) zhuhang@JUPITER:~$ cat /proc/version Linux version 5.3.0-40-generic (buildd@lcy01-amd64-024) (gcc version 7.4.0 (Ubuntu 7.4.0-1ubuntu1~18.04.1)) #32~18.04.1-Ubuntu SMP Mon Feb 3 14:05:59 UTC 2020 (base) zhuhang@JUPITER:~$ lspci | grep -i nvidia 02:00.0 VGA compatible controller: NVIDIA Corporation GP102 [GeForce GTX 1080 Ti] (rev a1) 02:00.1 Audio device: NVIDIA Corporation GP102 HDMI Audio Controller (rev a1) 82:00.0 VGA compatible controller: NVIDIA Corporation GP104 [GeForce GTX 1080] (rev a1) 82:00.1 Audio device: NVIDIA Corporation GP104 High Definition Audio Controller (rev a1)
-
根据系统版本及GPU型号寻找合适的显卡驱动并下载安装
wget -c https://us.download.nvidia.com/XFree86/Linux-x86_64/450.66/NVIDIA-Linux-x86_64-450.66.run
测试:输入
nvidia-smi
,如有表格输出说明驱动安装成功。
第二步 anaconda 安装
-
寻找anaconda的下载链接,
-
下载
wget https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/Anaconda3-2020.07-Linux-x86_64.sh
-
安装
bash Anaconda3-2020.07-Linux-x86_64.sh
-
更新环境变量
source ~/.bashrc
-
验证安装
(base) zhuhang@JUPITER:~$ python3 Python 3.8.3 (default, Jul 2 2020, 16:21:59) [GCC 7.3.0] :: Anaconda, Inc. on linux Type "help", "copyright", "credits" or "license" for more information.
第三步 建立tensorflow_gpu环境
建立环境,如下命令中py36tensor1为虚拟环境的名称(建议包括python版本和使用框架的名称),python=3.6为指定虚拟环境的python版本。
conda create -n py36tensor1 python=3.6
有关环境的常用命令
conda env remove -n [环境名称] //删除环境
conda env list //查看所有环境
conda activate [环境名称] // 激活环境
conda deactivate // 关闭环境
第四步 安装tensorflow
- 搜索指定包
conda search tensorflow-gpu
- 激活py36tensor1环境,并安装tensorflow-gpu指定版本
conda activate py36tensor1
conda install tensorflow-gpu=1.13.1
-
验证tensorflow安装
python -c "import tensorflow as tf;print(tf.reduce_sum(tf.random.normal([1000, 1000])))"
如果没有W开头的信息就代表安装成功。
注:如果想要安装conda没有而pip有的包,可以使用如下方式安装到当前虚拟环境下:
python -m pip install numpy