2024-05-16 08:30:28 发布
网友
有人知道如何在tensorflow 1.x版本中使用Google Colab/Colab Pro的GPU吗? 我尝试将我的CUDA从10.1降级到10.0(这不起作用)以及pip安装tensorflow gpu==1.14(建议不要这样做,因为这可能会降低性能)
要执行此操作,首先必须从GoogleColab卸载当前的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.list (Y/I/N/O/D/Z) [default=N] ?
Y
安装后,如果您可以检查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
!pip install tensorflow_gpu ==1.14
Google Colab使用%tensoflow magic:https://colab.research.google.com/notebooks/tensorflow_version.ipynb支持各种tensorflow版本
%tensorflow_version 1.x import tensorflow as tf print(tf.__version__, tf.test.is_gpu_available())
要执行此操作,首先必须从GoogleColab卸载当前的CUDA版本,如下所示
在第二步中,您可以下载CUDA 10.0并按如下所示进行安装
运行时,它会提示
*** cuda.list (Y/I/N/O/D/Z) [default=N] ?
,然后您必须提供Y
(Y或I:安装软件包维护程序的版本)安装后,如果您可以检查CUDA版本,如下所示
输出:
最后,您可以使用
pip
安装Tensorflow GPU,如下所示Google Colab使用%tensoflow magic:https://colab.research.google.com/notebooks/tensorflow_version.ipynb支持各种tensorflow版本
相关问题 更多 >
编程相关推荐