OneCode-cucldk.com

当前位置:首页 - 系统软件 - 正文

阿里云gpu服务器:

为了少踩坑,这里默认选择了自动安装GPU驱动,版本选9.0.176下的390.46;

ps:这里是在下单选操作系统的时候选择,后来打算重装系统的时候发现重装没有选驱动这一步,遂作罢未重装,不知道重装提交工单给不给装;系统开通后进去就能看到装驱动的log文件,成功即可;

系统更新:

yum update

更新Python3,参考https://cucldk.com/post/centos7-to-python3.html

https://developer.nvidia.com/cudnn官网下载cuDNN,需要注册账号登陆,下载版本 Download cuDNN v7.3.1 (Sept 28, 2018), for CUDA 9.0下的cuDNN v7.3.1 Library for Linux;

外国网站国内下载可能较慢,着急可用国外vps中转;

tar -zxvf cudnn-9.0-linux-x64-v7.3.1.20.tgz
cd cuda
cp include/* /usr/local/cuda-9.0/inlcude
cp lib64/lib* /usr/local/cuda-9.0/lib64

修改环境变量,执行 

sudo vim /etc/profile

在export PATH 那行下面加上两行代码:

export PATH=/usr/local/cuda-9.0/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64:$LD_LIBRARY_PATH

退出执行

source /etc/profile
pip3 install tensorflow-gpu==1.12.0

执行测试代码,看是否能够正常导入,并输出自定义的文字:

import tensorflow as tf

hello = tf.constant('Hello, TensorFlow!')
sess = tf.Session()
sess.run(hello)

之后执行

from tensorflow.python.client import device_lib
print(device_lib.list_local_devices())

检测到gpu即成功。

参考编辑:https://blog.csdn.net/Oh_My_Fish/article/details/78861867



本文来源:OneCode

本文地址:https://cucldk.com/post/aliyun-tensorflow-gpu-install.html

版权声明:如无特别注明,转载请注明本文地址!