我使用Google Colab来训练模型，但如图所示，当我输入“torch.cuda.is_available（）”时，输出为“true”。然后我运行代码，但出现RuntimeError错误：没有可用的CUDA GPU。
f87krz0w1#
尝试安装您要使用的cudatoolkit版本“conda install pytorch torchvision cudatoolkit=10.1 -c pytorch”
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数量，然后就工作了。我希望这也能解决你的问题。
2条答案
按热度按时间f87krz0w1#
尝试安装您要使用的cudatoolkit版本“conda install pytorch torchvision cudatoolkit=10.1 -c pytorch”
enxuqcxy2#
我想我已经找到了一个解决方案来修复我的错误。我曾经有过同样的错误。当你运行这个：
它会给予你GPU的编号，在我的例子中是
我意识到我传递的代码是：
所以我把“1”换成了“0”，也就是Colab给我的GPU数量，然后就工作了。
我希望这也能解决你的问题。