Amazon AWS上Tensorflow+GPU+CUDA 8+cuDNN 5+OpenBLAS配置

最新推荐文章于 2024-04-25 15:50:14 发布

huxuanlai

最新推荐文章于 2024-04-25 15:50:14 发布

阅读量1.2k

点赞数

分类专栏：数据挖掘和统计建模

本文链接：https://blog.csdn.net/huxuanlai/article/details/70256881

版权

//环境：Amazon AWS g2.2xlarge实例，Ubuntu 16.04, python3.5, Nvidia cuda 8, Tensorflow
//安装Python3和Tensorflow方法：
sudo apt-get install -y python3-pip
sudo pip3 install -y tensorflow
sudo pip3 install -y tensorflow-gpu
sudo pip3 install --upgrade \
https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow_gpu-1.0.1-cp35-cp35m-linux_x86_64.whl

//安装NVIDIA CUDA Toolkit 8.0
wget http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/cuda-repo-ubuntu1604_8.0.61-1_amd64.deb
sudo dpkg -i cuda-repo-ubuntu1604_8.0.61-1_amd64.deb
rm -rf cuda-repo-ubuntu1604_8.0.61-1_amd64.deb
sudo apt-get update
sudo apt-get install -y cuda

//安装NVIDIA cuDNN库
https://developer.nvidia.com/rdp/cudnn-download 注册下载 cudnn-8.0-linux-x64-v5.1.tgz
tar -xzvf cudnn-8.0-linux-x64-v5.0-ga.tgz
sudo cp -R cuda/lib64/lib* /usr/local/