Google Colab:更新cuda和tf库版本

更新cuda和库版本

Google默认安装的库一般都是最新的,例如tf2.0版本,但是我们想要用到tf1.0就会有诸多不便了,因此我需要更新cuda和tf-gpu以及tf-cpu版本。

安装cuda

在官网选择对应版本
Click~
Google Colab:更新cuda和tf库版本_第1张图片
安装cuda可以一步走完~

!wget https://developer.nvidia.com/compute/cuda/9.0/Prod/local_installers/cuda-repo-ubuntu1604-9-0-local_9.0.176-1_amd64-deb
!dpkg -i cuda-repo-ubuntu1604-9-0-local_9.0.176-1_amd64-deb
!apt-key add /var/cuda-repo-9-0-local/7fa2af80.pub
!apt-get update
!apt-get install cuda=9.0.176-1
#~~~

查看cuda版本

!nvcc -V

cat /usr/local/cuda/version.txt

安装cudnn

安装cudnn则不行,6.0之前的版本可以wget,7.0之后的版本要注册账号啥的,只能下载了文件《cudnn-9.0-linux-x64-v7.4.1.5.tgz》传进去,然后使用下面命令安装。

!sudo tar -zxvf cudnn-9.0-linux-x64-v7.4.1.5.tgz
!sudo cp cuda/include/cudnn.h /usr/local/cuda/include
!sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
!sudo chmod a+r /usr/local/cuda/include/cudnn.h
!sudo chmod a+r /usr/local/cuda/lib64/libcudnn*
#~~~

查看cudnn版本

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

安装对应的keras

!pip install keras==2.1.4

安装对应的tf

!pip install tensorflow-gpu==1.12
!pip install tensorflow==1.12

注意事项

安装完后可能会出现以下提醒,我们需要点击RESTART RUNTIME,相当于重启内核,更新我们刚刚安装的库, 不然刚刚安装的是无法生效的。
Google Colab:更新cuda和tf库版本_第2张图片

检查是否安装成功

import keras
keras.__version__
import tensorflow as tf
tf.__version__

卸载cuda可参考以下链接

Click~

sudo /usr/local/cuda-8.0/bin/uninstall_cuda_8.0.pl
sudo rm -rf /usr/local/cuda-8.0

卸载cudnn可参考以下链接

Click~

sudo rm -rf /usr/local/cuda/include/cudnn.h
sudo rm -rf /usr/local/cuda/lib64/libcudnn*

你可能感兴趣的:(实用小工具)