Pytorch GPU版本安装

一、背景

记录一下安装Pytorch GPU版本过程。

由于手残,卸载了电脑上的显卡驱动,现在我连显卡类型是啥都不知道了。

总体思路:安装显卡驱动->安装cuda->安装pytorch库

二、安装显卡驱动

2.1 查看本地显卡型号

通过「DirectX 诊断工具」查看显卡信息,以便下载驱动程序。

键盘win+r 输入 dxdiag 进入诊断工具界面

2.2 下载显卡驱动程序

进入人工智能计算领域的领导者 | NVIDIA英伟达官网,点击驱动程序。

根据自己电脑配置信息进行选择,然后点击搜索->下载。

下面进行安装,同意并继续->下一步->马上重启

cmd输入 nvidia-smi 可以看到驱动安装成功。注意下面驱动版本是552.22,后面的CUDA 版本是当前驱动支持的最高可以安装的版本

三、安装CUDA

3.1 判断所需CUDA版本

根据实际情况,我需要安装pytorch版本为torch==1.13.1。因为cuda、pytorch版本之间是有依赖性的,因此需要选择适合我们的CUDA版本。见官网Previous PyTorch Versions | PyTorch

找到了我需要的CUDA版本11.6或11.7。同时,此CUDA版本符合驱动能支持的最高CUDA版本,否则需要重新安装驱动。

接下来,去官网下载CUDA11.7CUDA Toolkit Archive | NVIDIA Developer

3.2 安装所需CUDA版本

检查是否安装成功,cmd中输入 nvcc -V

四、CUDNN安装(可选)

参考这篇文章GPU,CUDA,cuDNN的理解-CSDN博客,文章说CUDNN不是必须安装的。

我这里选择不安装。CUDNN下载地址cuDNN Archive | NVIDIA Developer。注意这个网站需

要注册(无语凝噎)

五、安装pytorch库

回到Previous PyTorch Versions | PyTorch选择自己要下载的torch版本。

我直接复制这个命令进行安装。

下载完成后验证。命令行输入 conda list

最后测试pytorch是否可以调用GPU,大功告成。

python 复制代码
# -*- coding:utf-8 -*-
# @FileName :test_GPU.py
# @Time :2024/4/28 14:21
# @Author :Tim
import torch
if torch.cuda.is_available():
    print("GPU is available")
else:
    print("GPU is not available")
相关推荐
振鹏Dong35 分钟前
依托 <AI 原生应用架构白皮书>,看 AI 原生应用的发展与实践
人工智能
智行众维2 小时前
自动驾驶的“虚拟驾校”如何炼成?
人工智能·自动驾驶·汽车·智能驾驶·智能网联汽车·智能驾驶仿真测试·智驾系统
空白到白2 小时前
NLP-注意力机制
人工智能·自然语言处理
大千AI助手4 小时前
指数分布:从理论到机器学习应用
人工智能·机器学习·参数估计·概率密度函数·mle·指数分布·累积分布函数
MATLAB代码顾问4 小时前
MATLAB绘制多种混沌系统
人工智能·算法·matlab
搬砖的小码农_Sky4 小时前
人形机器人:Tesla Optimus的AI集成细节
人工智能·ai·机器人
做运维的阿瑞4 小时前
2025 年度国产大模型「开源 vs. 闭源」深度评测与实战指南
人工智能·低代码·开源
渡我白衣4 小时前
深度学习入门(三)——优化算法与实战技巧
人工智能·深度学习
可触的未来,发芽的智生4 小时前
触摸未来2025.10.10:记忆的种子,当神经网络拥有了临时工作区,小名喜忆记系统
人工智能·python·神经网络·机器学习·架构
极客BIM工作室4 小时前
演化搜索与群集智能:五种经典算法探秘
人工智能·算法·机器学习