colab如何配置cuda9.1以及对应的cudnn和tensorflow-gpu
时间: 2024-06-10 19:06:38 浏览: 135
在服务器+cuda9.2+cudnn7.0可安装的tensorflow-gpu最新编译好的版本
感谢您的提问。关于如何配置Colab的cuda9.1以及对应的cudnn和tensorflow-gpu,以下是具体步骤:
1. 访问Colab网站(https://colab.research.google.com/);
2. 新建一个Notebook;
3. 在Notebook中输入以下命令进行安装:
```
!wget https://developer.download.nvidia.cn/compute/cuda/repos/ubuntu1604/x86_64/cuda-9-1_9.1.85-1_amd64.deb
!dpkg -i cuda-9-1_9.1.85-1_amd64.deb
!apt-get update
!apt-get install cuda=9.1.85-1
```
4. 安装cudnn:
```
!wget https://developer.download.nvidia.cn/compute/redist/cudnn/v7.6.5/cudnn-9.1-linux-x64-v7.6.5.32.tgz
!tar -zxvf cudnn-9.1-linux-x64-v7.6.5.32.tgz
!cp cuda/include/cudnn.h /usr/local/cuda-9.1/include
!cp cuda/lib64/libcudnn* /usr/local/cuda-9.1/lib64/
!chmod a+r /usr/local/cuda-9.1/include/cudnn.h /usr/local/cuda-9.1/lib64/libcudnn*
```
5. 安装tensorflow-gpu:
```
!pip install tensorflow-gpu==1.15.0
```
以上是配置Colab的cuda9.1以及对应的cudnn和tensorflow-gpu的详细步骤,希望对您有帮助。
阅读全文