安装CUDA以及GPU版本的pytorch

使用pytorch进行深度学习的时候,往往想用GPU进行运算来提高速度。于是搜索便知道了CUDA。

下面给出一个自检的建议:

  1. 检查cuda的版本是否适配自己的GPU。

打开NVDIA控制面板,点击左下角"系统信息",然后就可以看到NVDIA GPU的详细信息,其中就包含了CUDA的版本。在官网安装合适版本的cuda-toolkit。

  1. 安装了cuda,但是命令行输入nvcc -V 报错显示没有nvcc

这时候可能没有将CUDA添加到环境变量。检查系统变量中是否包含了CUDA_PATH ,以及CUDA_PATH_Vx.x , 以及PATH中是否包含了cuda的bin目录。

  1. 在命令行输入nvcc -V 可以正常运行,但是在python中使用print(torch.cuda.is_available()) 显示的是False。 这时候考虑torch安装错误。

(直接使用pip install torch 会安装cpu版本的torch,导致无法使用cuda)

卸载原先的torch,pip uninstall torch

然后在PyTorch官网找到正确版本进行下载安装。我的CUDA版本是12.6,PyTorch官网的最新版本支持CUDA12.4 安装命令如下:

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu124

此时在python中再运行print(torch.cuda.is_available())就会显示True了。

相关推荐
张较瘦_几秒前
[论文阅读] 人工智能 + 软件工程 | 需求获取访谈中LLM生成跟进问题研究:来龙去脉与创新突破
论文阅读·人工智能
一 铭1 小时前
AI领域新趋势:从提示(Prompt)工程到上下文(Context)工程
人工智能·语言模型·大模型·llm·prompt
云泽野4 小时前
【Java|集合类】list遍历的6种方式
java·python·list
麻雀无能为力4 小时前
CAU数据挖掘实验 表分析数据插件
人工智能·数据挖掘·中国农业大学
时序之心4 小时前
时空数据挖掘五大革新方向详解篇!
人工智能·数据挖掘·论文·时间序列
IMPYLH5 小时前
Python 的内置函数 reversed
笔记·python
.30-06Springfield5 小时前
人工智能概念之七:集成学习思想(Bagging、Boosting、Stacking)
人工智能·算法·机器学习·集成学习
说私域6 小时前
基于开源AI智能名片链动2+1模式S2B2C商城小程序的超级文化符号构建路径研究
人工智能·小程序·开源
永洪科技6 小时前
永洪科技荣获商业智能品牌影响力奖,全力打造”AI+决策”引擎
大数据·人工智能·科技·数据分析·数据可视化·bi
shangyingying_16 小时前
关于小波降噪、小波增强、小波去雾的原理区分
人工智能·深度学习·计算机视觉