cuda，pytorch显示False要如何解决

yoyami

我在安装完cuda后，运行python来检测返回False

import torch
print(torch.cuda.is_available())

要如何解决这个问题
我的配置如下

OmegaMolecule

如果是本机环境的话，装一下CUDA TOOLKIT就行了，如果是Docker的话run image记得加-gpuall参数

yoyami

Ω分子 发表于 2024-3-5 19:19
如果是本机环境的话，装一下CUDA TOOLKIT就行了，如果是Docker的话run image记得加-gpuall参数

我安装了CUDA TOOLKIT了，可是还是这样

OmegaMolecule

换个旧版的试试，比如11.8，11.6，12.2之类的，这12.4我还真没用过

freelive

https://www.cnblogs.com/alex-jzw/p/15021956.html

https://blog.csdn.net/qq_41112170/article/details/131191827

看下是否是版本不匹配的问题。

Bayonet

你这个安装得是CPU版本。
2.5.1+cu124  cu才是GPU，cpu是CPU

Bayonet

用这个网站的在线安装。https://pytorch.org/get-started/locally/