其他分享
首页 > 其他分享> > 如何安装 PyTorch

如何安装 PyTorch

作者:互联网

前言

现在使用 pytorch 框架进行深度学习的人越来越多了,但是 torch 的安装(GPU 版本)和普通第三方包的安装过程不太一样,因为涉及到电脑的显卡驱动版本,下面来看看安装 pytorch 的正确姿势。

安装过程

查看驱动版本

打开 NVIDIA 控制面板,点击工具栏帮助 --> 系统信息,在弹出的对话框中有如下信息:

系统信息

可以看到本机驱动程序版本为 466.92,接着前往 NVIDIA CUDA Toolkit Release Notes,在 cuda driver 这一节有下面这个表:

CUDA

由于本机为 Windows 系统且驱动版本为 466.92,所以应该选择 CUDA 11.6.x。

下载 torch

在 PyTorch 官网的 Get Started 页面有下图所示的安装向导:

安装向导

虽然没找到 CUDA 11.6.x,但是 CUDA 11.3 也是可以的,复制 Run this Command 里面的指令:

pip install torch==1.10.2+cu113 torchvision==0.11.3+cu113 torchaudio===0.10.2+cu113 -f https://download.pytorch.org/whl/cu113/torch_stable.html

这里把 pip3 换成了 pip,不过 pip3 也是可以的。在终端运行上述指令,等待安装完成即可。

标签:torch,cu113,pytorch,如何,PyTorch,CUDA,版本,安装
来源: https://www.cnblogs.com/zhiyiYo/p/15865454.html