您的位置:首页 > 其它

Ubuntu16.04+Anaconda3+TensorflowGPU+Cuda+Cudnn

2019-04-12 15:05 676 查看

Ubuntu16.04的安装问题就不多说了,直接从Anaconda3的安装开始:

1. 安装Anaconda

bash Anaconda3-2019.03-Linux-x86_64.sh

安装按照步骤来一步一步往下走即可,网上有很多教程,这里就不赘述。

安装Anaconda3以后记得建虚拟环境:

conda create -n MyEnvs python=3.7

删除Anaconda环境:

conda remove -n MyEnvs --all

2. 安装Cuda10.1,其对应的nvidia驱动是418(cuda版本对应驱动版本), 命令:

sudo apt-get install nvidia-418

sudo dpkg -i cudnn-10.1-linux-x64-v7.5.0.56.tgz

sudo apt-get update

sudo apt-get install cuda

3. 安装Cudnn,去官网下载对应的版本,这里应该安装Cudnn7.5

4. 安装Tensorflow-gpu

各版本的tensorflow-gpu与cuda和cudnn的版本对应关系:https://blog.csdn.net/omodao1/article/details/83241074

conda install tensorflow-gpu

安装的最新版本tensorflow-gpu==1.13版本,这个版本可以兼容python3.7和cuda10了,这也就是为什么我刚一开始就安装的cuda10.1和python3.7的原因

import tensorflow as tf

a = tf.constant(2)

b = tf.constant(3)

c = a * b

sess = tf.Session()

sess.run(c)    ###结果是6且终端信息有关于GPU的描述,在此说明tensorflow-gpu版本安装成功

5. 安装opencv3

用的如下命令

conda install -c https://conda.binstar.org/menpo opencv3

发现会自动安装Python2.7和tensorflow1.3的版本,覆盖之前的安装,所以后面采用:

pip install opencv-python

安装好用cv2.destroyAllWindwos()和cv2.__version__均没有问题。

 

内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: 
相关文章推荐