在 Google Colab 上从 CUDA 11.2 降级到 11.1 或 10.2(找不到包问题)

Posted

技术标签:

【中文标题】在 Google Colab 上从 CUDA 11.2 降级到 11.1 或 10.2(找不到包问题)【英文标题】:Downgrading from CUDA 11.2 to 11.1 or 10.2 on Google Colab (Package not found issue) 【发布时间】:2021-09-20 00:50:34 【问题描述】:

因此,由于 pytorch/pytorch-geometric 不支持它,因此我几乎尝试了所有我能想到的来降级 Google Colab (11.2) 上的 CUDA 版本。我已经遵循了有关该主题的多个其他帖子的建议,在线下载 CUDA 版本并使用(例如)擦除本地版本:

!wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-ubuntu2004.pin
!mv cuda-ubuntu2004.pin /etc/apt/preferences.d/cuda-repository-pin-600
!wget https://developer.download.nvidia.com/compute/cuda/11.1.0/local_installers/cuda-repo-ubuntu2004-11-1-local_11.1.0-455.23.05-1_amd64.deb
!dpkg -i cuda-repo-ubuntu2004-11-1-local_11.1.0-455.23.05-1_amd64.deb
!apt-key add /var/cuda-repo-ubuntu2004-11-1-local/7fa2af80.pub
!apt-get update
!apt-get -y install cuda = 11.1.0
!apt autoremove

但无论我是否尝试使用其他代码预先擦除 CUDA 版本、更改路径/名称或不同的 CUDA 版本,我都会得到相同的输出,即:

...
Note, selecting 'libghc-cryptohash-sha1-prof' instead of 'libghc-cryptohash-sha1-prof-0.11.100.1-c80f9'
E: Unable to locate package 

这是有关系统和 CUDA 的信息:

+-----------------------------------------------------------------------------+
| NVIDIA-SMI 460.32.03    Driver Version: 460.32.03    CUDA Version: 11.2     |
|-------------------------------+----------------------+----------------------+
| GPU  Name        Persistence-M| Bus-Id        Disp.A | Volatile Uncorr. ECC |
| Fan  Temp  Perf  Pwr:Usage/Cap|         Memory-Usage | GPU-Util  Compute M. |
|                               |                      |               MIG M. |
|===============================+======================+======================|
|   0  Tesla T4            Off  | 00000000:00:04.0 Off |                    0 |
| N/A   40C    P8    10W /  70W |      0MiB / 15109MiB |      0%      Default |
|                               |                      |                  N/A |
+-------------------------------+----------------------+----------------------+

No LSB modules are available.
Distributor ID: Ubuntu
Description:    Ubuntu 18.04.5 LTS
Release:    18.04
Codename:   bionic

对此的任何帮助将不胜感激,因为它会大大加快我现在只能在笔记本电脑上使用 CPU 的运行时间!

【问题讨论】:

【参考方案1】:

您遵循的步骤似乎是正确的。唯一的附加步骤是重新启动运行时(菜单->运行时->重新启动运行时)。 之后,检查

!nvcc --version

表演

nvcc:NVIDIA (R) Cuda 编译器驱动程序 版权所有 (c) 2005-2020 NVIDIA Corporation 建立于 Mon_Oct_12_20:09:46_PDT_2020 Cuda 编译 工具,版本 11.1,V11.1.105 构建 cuda_11.1.TC455_06.29190527_0

然而 nvidia-smi 显示相同的旧版本 11.2,它似乎并没有真正影响运行 PyTorch。

【讨论】:

以上是关于在 Google Colab 上从 CUDA 11.2 降级到 11.1 或 10.2(找不到包问题)的主要内容,如果未能解决你的问题,请参考以下文章

Google Colab 上的 PyTorch Geometric CUDA 安装问题

如何修复 google colab 上的 cuda 运行时错误?

在 google colab 中将 CUDNN 升级到 8.2

不使用多处理但在使用 PyTorch DataLoader 时在 google colab 上出现 CUDA 错误

CHECK(调用)函数在Google Colab中出错[重复]

Google Colab Fine Tuning BERT Base Cased with Transformers and PyTorch 中出现间歇性“RuntimeError: CUDA out