Ubuntu16.04配置GTX1070深度学习环境

Ubuntu16 + CUDA Toolkit 9.0 + cuDNN SDK v7 + Tensorflow1.6 + Keras2.1
首先要禁止自带的显卡驱动(我好像没管这步所以自行百度吧)
1. 下载CUDA
- https://developer.nvidia.com/cuda-release-candidate-download下载自己的版本,我下载的是cuda_9.0.176_384.81_linux.run.81_linux-run
- 进入tty1 Ctrl + Alt + F1
- 关闭图形化界面 sudo service lightdm stop
- 安装 sudo sh cuda_9.0.176_384.81_linux.run.81_linux-run 全部选默认此时会安装对应到显卡驱动,安装完成后会提醒你加入PATHLD_LIBRARY_PATH环境变量
- 打开图形化界面sudo service lightdm start
- 返回tty7图形界面Ctrl + Alt + F7
2. 下载cuDNN
- 我下载的cudnn-9.0-linux-x64-v7.tgz
- 覆盖,这里到名称要对应自己的版本,比如cudaX.X
- sudo cp -a cuda/include/cudnn.h /usr/local/cuda/include/
- sudo cp -a cuda/lib64/libcudnn* /usr/local/cuda/lib64/
- 这里可能还要添加一个环境变量,我记不清了
3. 安装Anaconda
- 略
4. 安装TensorFlow-GPU
- $ conda create -n tensorgpu pip python=3.6
- $ source activate tensorgpu
- (tensorgpu)$ pip install tensorflow-gpu==1.6
OK

猜你喜欢

转载自blog.csdn.net/u013982164/article/details/80341846