如果出现这种情况，可能安装的pytorch版本是cpu版本的，无法调用GPU。

解决方案

进入到pytorch的官网PyTorch

在选择版本页面中https://pytorch.org/get-started/previous-versions/

选择适合自己版本的包，复制下来即可

备注：nvidia-smi可以查看当前GPU最高支持的CUDA版本

nvcc -V即可查看CUDA Runtime API运行的版本

# pytorch # 人工智能 # python
Logo
2048 AI社区

有“AI”的1024 = 2048，欢迎大家加入2048 AI社区

更多推荐

cover

Kubernetes - kubelet 配置优化，提升节点资源调度效率

avatar 2048 AI社区
cover

Agent代码审查API能力与默认级别边界

avatar 2048 AI社区

C++ 第 13 课：值传递 —— 为什么函数里改了，外面却没变？

atest()test()66因为每次调用test()，都会重新创建一个新的a = 5，然后a++变成 6。3.因为a定义在外层main()里，内层if可以访问外层变量。4.b定义在if的{}里面，离开这对花括号后，它的作用域就结束了，所以外面不能直接使用。

avatar 2048 AI社区