查看电脑是否支持intel不做赘述;
一定要看好对应的cuda!!!!!
!!!!!!
提前搞好驱动和g++和gcc对应的cuda版本,可以在网上查看一下
!!!!!!
我是安装cuda10.0 + cudnn7.5 + Tensorflow-gpu 1.14(差不多可以跑所有tensorflow1.+版本程序)
一、Cuda下载安装(安装cuda10.0)
直接上连接CUDA10.0下载地址
https://developer.nvidia.com/cuda-10.0-download-archive?target_os=Linux&target_arch=x86_64&target_distro=Ubuntu&target_version=1804&target_type=runfilelocal
盗图一张
第二个拓展包可以不用添加,如向添加下载运行完cuda接着继续
若cuda安装包保存在:桌面
cd 桌面
sudo sh cuda_10.0.130_410.48_linux.run # sudo sh + 安装包地址
# sudo sh cuda_10.0.130.1_linux.run ### 可能会有用
然后直接按 ctrl+c 跳过协议,下面操作:
1.输入accept
Do you accept the previousily read EULA?
accept/decline/quit: accept
2.问是否安装驱动accept接下来的问题
选择:n 千万千万选择 n
剩下的选择y和Enter即可
配置环境:
打开终端输入 gedit ~/.bashrc
也可以vim ~/.bashrc
(我喜欢用第一个)
在文件最下面添加:记得点击右上角的保存
export PATH=/usr/local/cuda-10.0/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
export CUDA_HOME=/usr/local/cuda
注:若安装多版本CUDA需要将其中‘cuda-10.0’中的‘-10.0’删除
然后更新,并检测是否安装成功:
source ~/.bashrc
nvcc -V # nvcc --version
cat /usr/local/cuda/version.txt
正常出现cuda版本表示安装成功 /(或者)通过下面操作也可以
cd /usr/local/cuda/samples/1_Utilities/deviceQuery
sudo make
./deviceQuery
返回 Result = PASS 表示安装成功
二、安装cudnn
需要注册一个号用邮箱
下载地址
千万选择对应cuda版本的cudnn,例如我的选择cudnn7.5.0
然后点开对应的cudnn
方法一、可以点击cuDNN Library for Linux即可,下载后解压;
进入解压后的文件夹
在终端运行
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*
方法二、也可以选择(推荐方法一)
cuDNN Runtime Library for Ubuntu18.04 (Deb)
cuDNN Developer Library for Ubuntu18.04 (Deb)
cuDNN Code Samples and User Guide for Ubuntu18.04 (Deb)
三个都要下载(麻烦哦)
然后依次运行这三个包
sudo dpkg -i 地址包
操作完之后可以在终端看看cuDNN的版本:
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
cudn10.0 和cudnn7.5.0安装完成
三、安装Tensorflow
建议创建虚拟环境进行管理(以anaconda解释器pycharm为例)
# conda create -n [name] python.version 以我为例(版本号可以换)
conda create -n tf1 python=3.6.5
可能会出错超时,试试加–default-time==100(忘了一个=还是两个了)在pip和install之间
pip install tensorflow-gpu==1.14 -i https://pypi.tuna.tsinghua.edu.cn/simple
测试是否安装成功
在终端输入
python
返回结果true成功了,注:在输入import tensorflow后会出现一堆垃圾,不要在意
>>>import tensorflow
>>>tensorflow.test.is_gpu_available()
最后:如果想更换tensorflow版本可以直接下载安装就会替换原版本