CUDA版本PyTorch安装
安装驱动
检查CUDA版本,如果版本不对要重新安装下驱动
我的电脑显卡为 Nvidia的GTX950M,安装驱动版本517.48,这个版本的驱动为CUDA_11.7
移动版:GeForce Game Ready 驱动程序 | 517.48 | Windows 11 | NVIDIA
桌面版:GeForce Game Ready 驱动程序 | 517.48 | Windows 10 64-bit, Windows 11 | NVIDIA
要使用cuda,必须安装CUDA版本的PyTorch,且版本要和驱动的cuda版本匹配。可以执行以下语句检查当前是否支持CUDA:
import torch print(torch.cuda.is_available())
如果安装的是CPU版本的PyTorch或者CUDA版本不匹配,将打印False
下载cuda版PyTorch
Previous PyTorch Versions | PyTorch
# CUDA 11.7
pip install torch==1.13.1+cu117 torchvision==0.14.1+cu117 torchaudio==0.13.1 --extra-index-url https://download.pytorch.org/whl/cu117
在命令行下输入上面的语句
注意:之前已经安装了CPU版本的PyTorch也没事,这边会帮你自动卸载,然后替换为CUDA版本的PyTorch
再运行上面的语句,就打印True了
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】凌霞软件回馈社区,博客园 & 1Panel & Halo 联合会员上线
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】博客园社区专享云产品让利特惠,阿里云新客6.5折上折
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· DeepSeek “源神”启动!「GitHub 热点速览」
· 我与微信审核的“相爱相杀”看个人小程序副业
· 微软正式发布.NET 10 Preview 1:开启下一代开发框架新篇章
· C# 集成 DeepSeek 模型实现 AI 私有化(本地部署与 API 调用教程)
· spring官宣接入deepseek,真的太香了~