一招办理Pytorch GPU版本安装慢的问题

        Pytorch是一个盛行的深度学习框架，广泛应用于计算机视觉、自然语言处理等领域。安装Pytorch GPU版本可以充实利用GPU的并行计算本领，加速模子的训练和推理过程。接下来，我们将详细介绍如何在Windows操纵体系上安装Pytorch GPU版本。
检察是否有显卡

GPU版本的pytorch需要有显卡支持,如果没有显卡那就只能使用cpu版本了。
cpu版本安装：

1. pip install torch torchvision torchaudio

复制代码

 win+s搜索设备管理器，打开后，点击显示适配器

        若你的电脑有独立显卡且显卡版本GTX>10,RTX>20那么便可以使用GPU版本的Pytorch。
显卡版本小于上述版本的显卡是没有CUDA支持的,无法使用Pytorch的GPU版本，只能使用CPU版本。
检察CUDA版本 

 win+r cmd命令行里输入：

1. nvidia-smi

复制代码

检察CUDA版本：

我这里是12.6,记着这个版本号。
下载CUDA驱动 

NVIDIA驱动下载:
CUDA Toolkit Archive | NVIDIA Developer

https://developer.nvidia.com/cuda-toolkit-archive根据刚刚的CUDA版本选择,12.6就下载CUDA Toolkit 12.6即可

点击后,如许选择:

 然后按部就班安装即可。
Whl文件下载

        装过pytorch的都知道，用pip或conda命令从官网安装pytorch可能会出现runtime error等错误，安装成功后可能碰面临安装的是cpu版本的pytorch，或者是python版本与pytorch版本不匹配等问题。以是我发起可以从下面链接先下载好对应的whl文件，链接如下:
https://download.pytorch.org/whl/torch

https://download.pytorch.org/whl/torch

• cu126表现gpu cuda版本，下载好nvidia终端，并添加到情况变量后输入nvidia-smi  可以检察gpu版本号 
  
• cp312代表python版本号,必须和python版本同等,不然下载后上也无法安装。
  
• wind_amd64表现windows体系
  

按照自己的版本选择后点击即可在浏览器里下载

点击蓝色文件夹，跳转到下载目次
点击文件ctrl+shift+c或右键复制文件地点后，win+r输入cmd打开cmd命令行

输入pip install后将路径复制然后等候下载即可,一般来说1~2分钟即可下载完毕
验证

1. import torch
2. print(f"PyTorch版本: {torch.__version__}")
3. print(f"CUDA是可用的:{torch.cuda.is_available()}")
4. print(f"当前pytorch使用设备:{torch.device('cuda' if torch.cuda.is_available() else 'cpu')}")

复制代码

安装成功！

免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！更多信息从访问主页：qidao123.com:ToB企服之家，中国第一个企服评测及商务社交产业平台。