pytorch Google协作:Torch CUDA为真,但没有CUDA GPU可用

gudnpqoy  于 2022-11-09  发布在  Go
关注(0)|答案(2)|浏览(208)

我使用Google Colab来训练模型,但如图所示,当我输入“torch.cuda.is_available()”时,输出为“true”。然后我运行代码,但出现RuntimeError错误:没有可用的CUDA GPU。

f87krz0w

f87krz0w1#

尝试安装您要使用的cudatoolkit版本“conda install pytorch torchvision cudatoolkit=10.1 -c pytorch”

enxuqcxy

enxuqcxy2#

我想我已经找到了一个解决方案来修复我的错误。我曾经有过同样的错误。当你运行这个:

import tensorflow as tf
tf.test.gpu_device_name()

它会给予你GPU的编号,在我的例子中是

/device:GPU:0

我意识到我传递的代码是:

parser.add_argument('--gpu', type=str, default="1", help='choose GPU')

所以我把“1”换成了“0”,也就是Colab给我的GPU数量,然后就工作了。
我希望这也能解决你的问题。

相关问题