CUDA、pytorch、配置环境教程合集

1.CUDA卸载

参考:

【环境搭建】卸载&重新安装cuda+cudnn+anaconda+gpu版pytorch_卸载cuda-CSDN博客

【cuda】cuda安装与卸载升级_升级cuda-CSDN博客

检测代码:

https://www.zhihu.com/question/482335084

python 复制代码
import torch
print('CUDA版本:',torch.version.cuda)
print('Pytorch版本:',torch.__version__)
print('显卡是否可用:','可用' if(torch.cuda.is_available()) else '不可用')
print('显卡数量:',torch.cuda.device_count())
print('是否支持BF16数字格式:','支持' if (torch.cuda.is_bf16_supported()) else '不支持')
print('当前显卡型号:',torch.cuda.get_device_name())
print('当前显卡的CUDA算力:',torch.cuda.get_device_capability())
print('当前显卡的总显存:',torch.cuda.get_device_properties(0).total_memory/1024/1024/1024,'GB')
print('是否支持TensorCore:','支持' if (torch.cuda.get_device_properties(0).major >= 7) else '不支持')
print('当前显卡的显存使用率:',torch.cuda.memory_allocated(0)/torch.cuda.get_device_properties(0).total_memory*100,'%')

2.5070ti出现bug

参考:更新5070ti适配pytorch,解决pytorch不支持sm_120问题-陌上玉台咏新桑-陌上玉台咏新桑-哔哩哔哩视频

【5070TI启动Stable Diffusion时报pytorch capability sm_120 is not compatible ...的解决方案】_5070ti pytorch-CSDN博客

从40系显卡换5070TI后CUDA出现问题,解决方法_5070ti cuda版本-CSDN博客

【torch报错】RuntimeError: CUDA error: no kernel image is available for execution on torch解决方案_torch.cuda.deferredcudacallerror: cuda call failed-CSDN博客

出现错误(已解决)RuntimeError: CUDA error: no kernel image is available for execution on the device CUDA ker-CSDN博客

深度学习环境配置避坑-CUDA11.0+DGL1.8_dgl cuda 11.0-CSDN博客

最后的解决方案是:

CUDA12.8+pytorch2.7.0

可以向下兼容pytorch2.5.1版本的代码,成功跑通

3.相关官方网站

pytorch:

Start Locally | PyTorch

清华镜像源:

Index of /anaconda/cloud/pytorch/win-64/ | 清华大学开源软件镜像站 | Tsinghua Open Source Mirror

CUDA:

CUDA Toolkit - Free Tools and Training | NVIDIA Developer

相关推荐
糯米导航1 小时前
AI 视觉回归实战:截图对比不是“找不同”,如何让智能差异分析真正服务 UI 质量
人工智能·ui·回归
科技圈观察2 小时前
2026年好伴AI医疗专用大模型应用梳理与梯队参考
人工智能
jkyy20142 小时前
深耕AI健康医疗数据智库,赋能企业构建主动健康管理新生态
大数据·人工智能·健康医疗
cd_949217212 小时前
3D角色自动绑骨怎么做?用V2Fun完成建模、绑定、动作和导出
人工智能·3d
瑞禧生物tech2 小时前
SH-PEG-Biotin巯基-聚乙二醇-生物素 HS-PEG-Bio 深度解析
人工智能
向日的葵0063 小时前
langchain的Tools教程(三)
python·langchain·tools
冬奇Lab3 小时前
MCP 系列(06):MCP vs Function Calling——用数据说话的选型指南
人工智能
冬奇Lab3 小时前
每日一个开源项目(第159篇):Vibe-Trading - 用自然语言做量化研究,AI 驱动的个人交易 Agent
人工智能·开源·资讯
AI大模型-小雄3 小时前
2026 年 7 月国内怎么开通 ChatGPT Pro?5x / 20x 区别、适合人群与避坑指南
人工智能·gpt·chatgpt·ai编程·开发工具·codex·chatgpt pro
快乐非自愿3 小时前
AI低代码破局:数字化降本增效的核心逻辑与商业落地
人工智能·低代码