【debug记录】有gpu,但是 pytorch仍显示 cpu【原来是新电脑没安装cuda】

原来是新电脑没安装cuda,以为安装了pytorch包就可以了。

检查过程:

  1. 'nvcc' 不是内部或外部命令,也不是可运行的程序, 说明没有安装cuda。

  2. 查看电脑显卡最高支持cuda版本:nvidia-smi

  3. 安装cuda,选择版本:https://developer.nvidia.com/cuda-toolkit-archive

    查看安装的版本:nvcc -V

  4. 安装pytorch gpu版本:https://pytorch.org/

  5. 查看 pytorch gpu 是否安装成功, 显示True则成功。

python 复制代码
import torch
print(torch.__version__)
print(torch.cuda.is_available())
相关推荐
ATM0061 小时前
人机协作系列(四)AI编程的下一个范式革命——看Factory AI如何重构软件工程?
人工智能·大模型·agent·人机协作·人机协同
读创商闻2 小时前
极狐GitLab CEO 柳钢——极狐 GitLab 打造中国企业专属 AI 编程平台,引领编程新潮流
人工智能·gitlab
kailp2 小时前
语言模型玩转3D生成:LLaMA-Mesh开源项目
人工智能·3d·ai·语言模型·llama·gpu算力
marteker2 小时前
弗兰肯斯坦式的人工智能与GTM策略的崩溃
人工智能·搜索引擎
无心水2 小时前
大语言模型零样本情感分析实战:无需机器学习训练,96%准确率实现指南
人工智能·机器学习·语言模型
来自于狂人2 小时前
AI大模型训练的云原生实践:如何用Kubernetes指挥千卡集群?
人工智能·云原生·kubernetes
橡晟8 小时前
深度学习入门:让神经网络变得“深不可测“⚡(二)
人工智能·python·深度学习·机器学习·计算机视觉
墨尘游子8 小时前
神经网络的层与块
人工智能·python·深度学习·机器学习
Leah01058 小时前
什么是神经网络,常用的神经网络,如何训练一个神经网络
人工智能·深度学习·神经网络·ai
倔强青铜38 小时前
苦练Python第18天:Python异常处理锦囊
开发语言·python