Pycharm虚拟Python环境中装TensorFlow GPU版

  • A+
所属分类:机器学习
摘要

查看CUDN版本,Pycharm中新建基于python 2.7的 Virtualenv,用pip安装TensorFlow-GPU

背景

我一直使用实验室的电脑,GPU是TITAN X(pascal),系统是Ubuntu 16.04,TensorFlow等软件是师兄师姐已经装好的。最近在看一篇论文,论文的源码已经在GitHub上开源了,想复现一下。但其代码所用Python版本是Python 2.7,而实验室电脑的TensorFlow是基于Python 3的,所以需要重新在Python 2.7环境下安装TensorFlow。

1.了解运行环境

由于不同版本的TensorFLow对应不同的CUDN,所以要提前知道电脑已安装CUDN的版本,以免出现不兼容的情况。

查看CUDN版本:

查看CUDNN版本:

查看已安装TensorFlow版本:

在py文件中输入并运行:

2.在Pycharm中新建基于Python 2.7的Virtualenv

新建一个Project,在Project Interpreter中进行如图所示操作

Pycharm虚拟Python环境中装TensorFlow GPU版

3.安装TensorFlow-GPU Package

在Pycharm setting中的Project Interpreter中可以安装Package,但速度奇慢,所以我选择使用

Virtualenv所对应的pip来安装TensorFlow。

首先找到此pip:Project路径/venv/bin

在此路径下运行

Pycharm虚拟Python环境中装TensorFlow GPU版

注意:./pip install tensorflow-gpu 此命令只会安装最新版本TensorFlow,但最新版本不一定与你安装的CUDN版本匹配,一定要指明所安装TensorFlow版本。

其他比较大的Package也可通过此方式安装,速度很快。

许龙涛

发表评论

:?: :razz: :sad: :evil: :!: :smile: :oops: :grin: :eek: :shock: :???: :cool: :lol: :mad: :twisted: :roll: :wink: :idea: :arrow: :neutral: :cry: :mrgreen: