Pytorch GPU版本安装全过程
如何查看当前 PyTorch 版本，在你的项目中运行如下代码（如果没有安装过，则跳过）
import torch
print(f"PyTorch 版本: {torch.__version__}")
# 如果输出 PyTorch 版本: 版本号+cpu
说明你的 PyTorch 是 CPU 版本，如果想使 GPU 加速代码，请跟我下面的步骤来
首先检查你的电脑是否有英伟达显卡（NVIDIV）
一般笔记本可以查看在笔记本C面上是否有一张绿色的小贴纸，上面如果有 NVIDIV 可直接去下一步
如果时间长或者没贴纸的也别着急，可以跟着我下面的步骤检查。
运行：Win + R 打开运行对话框，输入 devmgmt.msc 然后按回车
就可以进入到设备管理器页面，在此页面打开显示适配器，GPU 前带 NVIDIA就没问题了。
一、查看电脑显卡驱动版本
打开命令提示符（Windows）或终端（macOS/Linux）
输入以下命令并回车：
nvidia-smi
在输出的信息右上角，你会看到一行 CUDA Version: xx.x。这个数字是你的显卡驱动最高支持的 CUDA 版本。
这里我的版本是13.2
二、安装
1. 卸载PyTorch
如果只前学习安装过 cpu 版本的 PyTorch 的话,，我们需要先卸载原先的 PyTorch （若未安装过，则进行下一步）。
在终端中运行以下代码（如果安装到虚拟环境中，则需要先进入虚拟环境，再运行下面代码）
pip uninstall torch torchvision torchaudio
2. 安装GPU版本的PyTorch
建议 本机的CUDA 版本 >= 安装的CUDA版本，例如我的版本是13.2 > 12.6 就可以
可以去往PyTorch官网，进入主页往下滑，看到如下页面 Install PyTroch，根据你的情况选择
然后打开终端运行它给出的代码（若有项目则在项目虚拟环境终端运行）
例如：
pip3 install torch torchvision --index-url https://download.pytorch.org/whl/cu126
三、检验
1. 终端检验
在终端中运行 pip list （若有项目则在项目虚拟环境终端运行）
看到版本号后加cu和刚才你安装版本号一致就安装成功
2. 代码检验
输出 True，则说明安装无误
import torch print(torch.__version__) print(torch.version.cuda) print(torch.cuda.is_available()) #输出为True，则安装无误
总结
以上为个人经验，希望能给大家一个参考，也希望大家多多支持脚本之家。
相关文章
今天小编就为大家分享一篇Python 限定函数参数的类型及默认值方式，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2019-12-12
这篇文章主要介绍了Python延迟绑定问题原理及解决方案,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下2020-08-08
缺失值是指数据集中的某些观察存在遗漏的指标值,缺失值的存在同样会影响到数据剖析和挖掘的效果,下面这篇文章主要给大家介绍了关于Python Pandas中缺失值NaN的判断,删除及替换的相关资料,需要的朋友可以参考下2022-01-01
本文主要介绍了Pycharm远程连接服务器跑代码的实现，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2022-07-07
这篇文章主要介绍了python写入中英文字符串到文件的方法,实例分析了Python操作中英文字符串的技巧,非常简单实用,需要的朋友可以参考下2015-05-05
模拟键盘操作执行自动化任务，我们常用的有pyautowin等自动化操作模块。今天介绍的这个模块叫做keyboard,它是纯Python原生开发，编译时完全不需要依赖C语言模块。一行命令就能完成安装，非常方便，需要的可以了解一下2022-10-10
最近的项目需要根据用户所属时区制定一些特定策略，学习、应用了若干python3的时区转换相关知识，具有一定的参考价值，感兴趣的小伙伴们可以参考一下2021-08-08
在本篇文章中小编给大家整理的是一篇关于go语言和python的区别点，需要的朋友们可以学习下。2020-07-07
Python+Tkinter实现Windows Hosts文件编辑管理工具
在日常开发和网络调试或科学上网场景中,Hosts文件修改是每个开发者都绕不开的"必修课",本文将完整解析一个基于Python Tkinter开发的Windows Hosts管理工具,希望对大家有一定的帮助2025-05-05
队、栈和链表一样，在数据结构中非常基础一种数据结构，同样他们也有各种各样、五花八门的变形和实现方式。这篇文章主要介绍了Python 实现数据结构-堆栈和队列的操作方法,需要的朋友可以参考下2019-07-07
最新评论