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了

 

posted @   yanghui01  阅读(175)  评论(0编辑  收藏  举报
相关博文:
阅读排行:
· DeepSeek “源神”启动!「GitHub 热点速览」
· 我与微信审核的“相爱相杀”看个人小程序副业
· 微软正式发布.NET 10 Preview 1:开启下一代开发框架新篇章
· C# 集成 DeepSeek 模型实现 AI 私有化(本地部署与 API 调用教程)
· spring官宣接入deepseek,真的太香了~
点击右上角即可分享
微信分享提示