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

相关推荐
AI行业学习2 分钟前
Claude Code + cc-switch + Git + Node.js 一站式完整安装配置教程(2026最新·国内可用版)
人工智能·git·python·安全·node.js·html·notepad++
denggun123454 分钟前
大大模型 如何通过 本地排雷 远端GPU
python·embedding·pip
汽车仪器仪表相关领域6 分钟前
NHA-604/605汽车排放气体测试仪:00级国标计量|多动力车型兼容|移动/固定双工况|智能联网尾气检测设备
大数据·人工智能·功能测试·安全·汽车·压力测试·可用性测试
科技园长8 分钟前
睿视医疗与晓闻科技达成战略合作,推动宠物专用CT设备智能化、数字化全面升级
人工智能·科技·宠物
杰瑞学AI11 分钟前
一个回答需要10分钟:飞书问答机器人踩坑实录——纯Agent自由检索,差点让我们的机器人“难产”
人工智能·机器人·prompt·transformer·飞书·ai-native
灵枢时代14 分钟前
小程序在生鲜配送行业,如何设计夜间预约和次日达的订单处理流程?
数据结构·人工智能·小程序
June`15 分钟前
常量内存和只读缓存
c++·人工智能·算法·cuda
IT小盘19 分钟前
16-Prompt版本管理-从手工修改到可追踪配置系统
开发语言·人工智能·python·prompt
卷无止境20 分钟前
Python进程池那些事儿:从原理到实战
后端·python
满怀冰雪20 分钟前
15-Paddle 高层 API 入门:paddle.Model 的训练与评估流程
人工智能·python·深度学习·机器学习·paddle