有谁知道如何将 Google Colab/Colab Pro 的 GPU 用于 TensorFlow 1.x 版本? 我尝试将 CUDA 从 10.1 降级到 10.0(这不起作用)以及 pip 安装tensorflow gpu==1.14(建议不要这样做,因为可能会降低性能)
最佳答案
要执行此操作,首先您必须从 Google Colab 卸载当前的 CUDA 版本,如下所示
!apt-get --purge remove cuda nvidia* libnvidia-*
!dpkg -l | grep cuda- | awk '{print $2}' | xargs -n1 dpkg --purge
!apt-get remove cuda-*
!apt autoremove
!apt-get update
第二步,您可以下载CUDA 10.0并安装,如下所示
!wget --no-clobber https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/cuda-repo-ubuntu1804_10.0.130-1_amd64.deb
!dpkg -i cuda-repo-ubuntu1804_10.0.130-1_amd64.deb
!sudo apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/7fa2af80.pub
!apt-get update
!apt-get install cuda-10-0
运行时提示*** cuda.list (Y/I/N/O/D/Z) [default=N] ?
,那么你必须提供 Y
(Y 或 I:安装软件包维护者的版本)
安装完成后,是否可以查看CUDA版本,如下图
!nvcc --version
输出:
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2018 NVIDIA Corporation
Built on Sat_Aug_25_21:08:01_CDT_2018
Cuda compilation tools, release 10.0, V10.0.130
最后,您可以使用pip
安装Tensorflow GPU,如下所示
!pip install tensorflow_gpu ==1.14
关于python - 谷歌 Colab GPU Tensorflow 1.x,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/62584399/