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

 

5. set gpu

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

export CUDA_VISIBLE_DEVICES=3

相关文章:

  • 2022-12-23
  • 2022-12-23
  • 2021-12-03
  • 2021-12-03
  • 2021-12-05
  • 2021-10-04
  • 2021-12-24
猜你喜欢
  • 2021-05-24
  • 2021-12-04
  • 2022-03-03
  • 2021-06-12
  • 2021-08-16
  • 2021-06-20
  • 2022-12-23
相关资源
相似解决方案