【Tensorflow】Tensorflow r1.0, Ubuntu, gpu, conda安装说明

Install Anaconda and python

  1. cuda-8.0

download cuda_8.0.61_375.26_linux.run

./cuda_8.0.61_375.26_linux.run

2.cudnnv5.1 for cuda-8.0

download cudnn-8.0-linux-x64-v5.1.tgz

tar xvzf cudnn-8.0-linux-x64-v5.1.tgz

cp cuda/include/cudnn.h cuda-8.0/include

cp cuda/lib64/libcudnn* cuda-8.0/lib64

  1. set path

vi .bashrc

source .bashrc

nvcc –version

4.tensorflow

conda create -n tensorflow

source activate tensorflow

(tensorflow)$ pip install –ignore-installed –upgrade TF_PYTHON_URL

TF_PYTHON_URL for python2.7: https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow_gpu-1.0.0-cp27-none-linux_x86_64.whl

  1. set gpu

http://www.acceleware.com/blog/cudavisibledevices-masking-gpus

export CUDA_VISIBLE_DEVICES=3

Original: https://www.cnblogs.com/ganganloveu/p/6404979.html
Author: 陆草纯
Title: 【Tensorflow】Tensorflow r1.0, Ubuntu, gpu, conda安装说明

原创文章受到原创版权保护。转载请注明出处:https://www.johngo689.com/9562/

转载文章受原作者版权保护。转载请注明原作者出处!

(0)

大家都在看

最近整理资源【免费获取】:   👉 程序员最新必读书单  | 👏 互联网各方向面试题下载 | ✌️计算机核心资源汇总