ubuntu16.04配置tensorflow-gpu并导入pycharm过程及遇到的问题集合

本文详细介绍了在Ubuntu16.04上配置tensorflow-gpu 1.4.0及cuda8.0、cudnn6.0的过程,包括环境变量配置和解决ImportError问题。同时,指导了如何将tensorflow导入pycharm,涉及python3路径设置和项目解释器配置。
摘要由CSDN通过智能技术生成

首先需要说明的是,在安装这种复杂的环境时,最好把之前所有涉及到与本次相关的软件全部卸载。不然会出现版本不兼容,多个残缺的版本存在互相干扰的情况。而在这种情况下,排除问题的难度时耗时又耗力的。

一、用到的软件说明

1、使用的软件版本号详情:

python 3.5.2
tensorflow-gpu 1.4.0
cuda 8.0. #使用nvcc -V 查看
cudnn 6.0

说明:
tensorflow 1.2对应cuda8.0 + cudnn 5.1
tensorflow 1.3以及1.4对应cuda8.0 + cudnn6.0

2、各软件版本号查看方式:

python :

python3 #这样输出的是你自己安装的python版本

tensorflow-gpu:

python3  #首先进入python环境
import tensor flow as tf #引入tensorflow组件
tf.__version__ #是两个下划线,查看版本
tf.__path__  #查看tensorflow安装位置

cuda:

nvcc -V

cudnn:

sudo cat /usr/local/cuda-8.0/include/cudnn.h | grep CUDNN_MAJOR -A2 #这里是我的cudnn的安装路径

二、环境配置

首先的一个安装流程是cuda->cudnn->tensorflow,当然前提是python的环境已经配置好。

1、安装cuda8.0

参考如下博,虽然博主是基于ubuntu14.04安装的,但是方法都是一样的:

cuda安装和测试点击这里

在这里我主要想提一下在配置cuda时遇到的问题:
1)一定要牢记的是cuda的安装路径在

/usr/local/cuda-8.0

2)配置环境变量,添加cuda路径到bashrc文件,其实这里有两种方式:
(1

  • 1
    点赞
  • 9
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值