首页 > 系统服务 > 详细

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

时间:2017-02-16 14:11:51      阅读:893      评论:0      收藏:0      [点我收藏+]

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

 

3. set path

vi .bashrc

----- add to end

export CUDA_HOME=[path for cuda-8.0]

export LD_LIBRARY_PATH=${CUDA_HOME}/lib64:${CUDA_HOME}/extras/CUPTI/lib64

-----

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

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

原文:http://www.cnblogs.com/ganganloveu/p/6404979.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!