gpt4 book ai didi

python - torch.cuda.is_available() 在 colab 中返回 false

转载 作者:行者123 更新时间:2023-12-04 00:23:01 39 4
gpt4 key购买 nike

我正在尝试在 google colab 中使用 GPU。以下是我的 colab 中安装的 pytorch 和 cuda 版本的详细信息。

Torch 1.3.1 CUDA 10.1.243

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

我对使用 GPU 在 pytorch 模型上进行迁移学习还很陌生。我的 torch.cuda.is_available() 返回 false,我无法使用 GPU。 torch.backends.cudnn.enabled 返回 true。这里可能出了什么问题?

最佳答案

确保您的硬件加速器设置为 GPU。
Runtime > Change runtime type > Hardware Accelerator

关于python - torch.cuda.is_available() 在 colab 中返回 false,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/59154623/

39 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com