有个单卡的服务器,想在上面装tensorflow 1.0,但是报错提示需要cuda 8.0,而本机的cuda是7.0.
于是就想进行升级:
文章图片
升级之前的准备工作
1uname -m && cat /etc/*release# get ubuntu version
2uname -r# get ubuntu kernal version
3gcc --version# get gcc version
4lspci | grep -i nvidia# check nvidia GPU information
5cat /proc/driver/nvidia/version# check nvidia GPU driver version
在https://developer.nvidia.com/cuda-downloads下载了对应的安装包。
然后执行安装命令:
文章图片
过了一小段时间,大约几分钟吧,安装完成:
文章图片
之后,执行如下命令:
apt-get update
apt-get install cuda
安装完成后,执行如下命令:
sudo vi ~/.bashrc# add PATh and LD_LIBRARY_PATH at the end
export PATH=/usr/local/cuda-8.0/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64:$LD_LIBRARY_PATH
Source ~/.bashrc
sudo reboot
重启后尝试:
dpkg -l | grep nvidia
nvidia-smi
cd/usr/local/cuda-8.0/samples
make# compile cuda-8.0 sample files
cd bin/x86_64/linux/release/# run sample file
./deviceQuery
./bandwidthTest
【Tensorflow|cuda由7.0升级到8.0】到此升级cuda到8.0结束。
推荐阅读
- Keras|将Pytorch模型迁移到android端(android studio)【未实现】
- Tensorflow|Tensorflow学习笔记----梯度下降
- Tensorflow【branch-官网代码实践-Eager/tf.data/Keras/Graph】_8.19
- nlp|Keras(十一)梯度带(GradientTape)的基本使用方法,与tf.keras结合使用
- tensorflow|tf1.x究竟到底如何如何使用Embedding?
- python|Keras TensorFlow 验证码识别(附数据集)
- AI|bert实现端到端继续预训练
- tensorflow|利用Tensorflow的队列多线程读取数据
- 深度学习|conda源,tensorflow2,pytorch安装