手把手教你安装 pytorch–GPU 版 1050Ti

1.检查你的电脑是否有 NVIDIA 显卡

打开任务管理器 >> 性能 可以看到右上角的 NVIDIA 以及显卡型号字样

pytorch lightnin 多显卡 pytorch支持显卡_命令行

2.在官网查看自己的显卡是否支持 CUDA 安装

NVIDIA官网地址:https://developer.nvidia.com/cuda-gpus

pytorch lightnin 多显卡 pytorch支持显卡_显卡驱动_02

我的 NVIDIA 显卡型号是 1050Ti ,没有显示在官网上,但实际上是支持 CUDA 的。

3.查看自己的显卡驱动版本

命令行输入以下命令

NVIDIA-SMI

pytorch lightnin 多显卡 pytorch支持显卡_官网_03


查看显卡驱动版本

pytorch lightnin 多显卡 pytorch支持显卡_命令行_04

4.更新最新版显卡驱动

大多数小伙伴的显卡驱动版本应该都没有更新到最新版,想要使用高版本的 cuda toolkit 需要更新高版本的驱动,这里我们先更新到最新版。

NVIDIA 显卡驱动官网地址:https://www.nvidia.com/Download/index.aspx?lang=cn

pytorch lightnin 多显卡 pytorch支持显卡_显卡驱动_05


搜索后直接点击下载即可。

pytorch lightnin 多显卡 pytorch支持显卡_官网_06


pytorch lightnin 多显卡 pytorch支持显卡_pytorch_07


这里使用默认路径就行。后面的安装也一直默认,下一步就可以了,不需要更改。

在命令行中输入以下代码,查看是否更新成功。

NVIDIA-SMI

pytorch lightnin 多显卡 pytorch支持显卡_命令行_08

5.到 pytorch 官网选择 pytorch 版本

pytorch lightnin 多显卡 pytorch支持显卡_pytorch_09


注意 Compute Platform 的 CUDA 版本不要大于命令行显示的 CUDA Version.

其实原本我是使用 conda 方式下载,但是镜像源出了问题,用了很多方式都无解,最后使用 pip 方式下载了。

pytorch lightnin 多显卡 pytorch支持显卡_显卡驱动_10

命令行输入命令进行下载(在此之前,可以先选择好虚拟环境)

pytorch lightnin 多显卡 pytorch支持显卡_pytorch_11


由于我已经安装过了,所以输入后会出现 Requirement already satisfied. 没有安装过的小伙伴只需要等待下载即可。安装完成后,命令行输入 conda list (安装了anaconda的) 或 pip list,查找需要下载的这三个包是否已经在 list 中了。使用 conda 命令安装的则查看对应 conda 命令下载的包。

pytorch lightnin 多显卡 pytorch支持显卡_pytorch_12

5.检查 pytorch 是否可用

打开 python 编译器,新建 python 文件,输入以下代码并运行,如果返回 True,则 pytorch 安装完成。

import torch
print(torch.cuda.is_available())

pytorch lightnin 多显卡 pytorch支持显卡_pytorch_13

Pytorch 安装到此就完成啦!