1、更新NVIDIA驱动
基于ppa源安装NVIDIA驱动
1 sudo add-apt-repository ppa:graphics-drivers/ppa
2 sudo apt-get update
此时打开Software & Updates ,选择 Additioal Drivers,选择要安装的驱动版本,点击右下角的 Apply Changes按钮。
(因为我已安装了对应版本,所以 Apply Changes按钮不可点,未安装时是可点的)
耐心等待,安装完成后,点击 Revert 按钮(重启电脑)
注意:python3.6对应的Anaconda版本是:Anaconda3-5.2.0-Linux-x86_64.sh 。此版本的Anaconda安装 tensorflow-gpu 时自动安装的CUDA版本是9.2 。依据下表,故NVIDIA的版本需 。
2 cuda cuddnn 安装
(1) 安装cuda
设置环境变量,在/home/
根据你的安装位置和版本
export PATH=$PATH:/usr/local/cuda-9.2/bin
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-9.2/lib64
(2) 安装cudnn
解压下载的文件,可以看到cuda文件夹,在当前目录打开终端,执行如下命令:
sudo cp cuda/include/cudnn.h /usr/local/cuda/include/
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/
sudo chmod a+r /usr/local/cuda/include/cudnn.h
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*
(3) 查看cudnn版本
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
3、安装Anaconda
去Anaconda官网下载所需的版本,本人下载的时python3.6对应的版本。(python3.6对应的Anaconda版本是:Anaconda3-5.2.0-Linux-x86_64.sh python3.5对应的Anaconda版本是:Anaconda3-4.2.0-Linux-x86_64.sh )(tensorflow应该暂不支持python3.7)不同Anaconda版本的安装的tensorflow版本不同(前者越新,安装的tensorflow版本越新),其对应的cuda版本也不同,特别注意版本对应的问题。
打开终端,输入如下命令进行安装
bash Anaconda3-5.2.0-Linux-x86_64.sh
中间会有“是否添加环境变量”的选项(如下图),选择 yes
4、使用 Conda 安装tensorflow-gpu
其中将第一句改为:conda create --name tftest python=3.6
第3句改为:conda install tensorflow-gpu
耐心等待安装完成(这个过程比较漫长,慢慢等待)
5、测试tensorflow是否安装成功
可以通过一个简单的程序测试TensorFlow是否安装成功。直接在终端中键入以下代码:
import tensorflow as tf
sess = tf.Session()
a = tf.constant(1)
b = tf.constant(2)
print(sess.run(a+b))
结果:3
6、PyCharm的配置
打开PyCharm之后,点击右下角的Configure,选择Settings,在左侧选择Project Interpreter,然后点击右侧面板的小齿轮,选择Add...,在弹出的窗口左侧选择Conda Environment,接着选择Existing environment,这时候它会自动识别出刚刚创建好的conda环境,然后勾选Make available to all projects,点击OK即可。
创建新项目的时候注意选择好Project Interpreter,因为刚刚的设置可能会不能应用到新项目,需要手动选择一下。
7、环境中opencv安装:
(1)sudo apt-get install python3-pip
(2)升级pip3)
其实升级pip3只需要两步,1.输入升级指令 2.修改配置文件。
升级指令为:
sudo pip3 install --upgrade pip
注意:如果只进行这一步的话接下来的安装仍会报错,提示错误大致为:cannot import name 'main'。所以还需要修改/usr/bin/pip3文件。
先通过 sudo vim /usr/bin/pip3 打开文件,并将文件修改为
from pip import __main__
if __name__ == '__main__':
sys.exit(__main__._main())
保存退出后即可完成pip3的更新。/usr/bin/pip3文件如下图所示:
(3)
pip install opencv-python
7、pip 换源
1)检查pip.conf文件是否存在
>> cd ~
>> mkdir .pip
>> ls ~/.pip
2)直接编辑pip.conf
sudo gedit ~/.pip/pip.conf
[global]
index-url = http://mirrors.aliyun.com/pypi/simple/
[install]
trusted-host = mirrors.aliyun.com