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判断文本中消息重复次数的方法,涉及Python针对文本文件的读取与字符串操作的相关技巧,需要的朋友可以参考下2016-04-04
这篇文章主要为大家详细介绍了python文件和文件夹复制函数的实现代码，文中示例代码介绍的非常详细，具有一定的参考价值，感兴趣的小伙伴们可以参考一下2020-02-02
这篇文章主要介绍了pycharm sql语句警告的处理方式，具有很好的参考价值，希望对大家有所帮助。如有错误或未考虑完全的地方，望不吝赐教2022-07-07
这篇文章主要为大家详细介绍了Python实现字符串的逆序，C++将字符串逆序输出，文中示例代码介绍的非常详细，具有一定的参考价值，感兴趣的小伙伴们可以参考一下2018-04-04
在Python中处理经纬度信息,首先通过筛选得到符合条件的停车点,并去除重复数据,利用这些数据点的索引值将原始数据进行分段,并绘制地图,处理重复数据时可以使用set()方法或列表生成式,最后,可以通过DataFrame的join方法合并具有相同索引值的数据2024-09-09
如何解决安装包过程中的Requirement already satisfied:问题
这篇文章主要介绍了如何解决安装包过程中的Requirement already satisfied:问题,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教2023-11-11
这篇文章主要介绍了如何解决PyTorch程序占用较高CPU问题,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教2023-09-09
这篇文章主要介绍了Python中使用Boolean操作符做真值测试实例,在Python中，任何类型的对象都可以做真值测试，并且保证返回True或者False,需要的朋友可以参考下2015-01-01
这篇文章主要介绍了Django 视图层(view)的使用，详细的介绍了什么是视图层以及案例，非常具有实用价值，需要的朋友可以参考下2018-11-11
这篇文章主要介绍了python lxml中etree的简单应用，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2019-05-05
最新评论