Manjaro下安装Pytorch手记

1.在软件安装器里查找pytorch,我需要安装CUDA加速的,选择下面python-pytorch-cuda这个包:

Manjaro下安装Pytorch手记_第1张图片

2.在manjaro setting里面检查系统驱动版本,可以看到我的是440xx这个驱动:

Manjaro下安装Pytorch手记_第2张图片

3.在安装管理器里面的选择中选择自己系统对应的显卡硬件驱动即可:

Manjaro下安装Pytorch手记_第3张图片

4.安装玩torch之后记得也把torchvision安装了:

Manjaro下安装Pytorch手记_第4张图片

5.然后在终端进入python,执行以下命令,没有报错说明已经安装好了:

Python 3.8.1 (default, Jan 22 2020, 06:38:00)
[GCC 9.2.0] on linux
Type "help", "copyright", "credits" or "license" for more information.

import torch
import torchvision

你可能感兴趣的:(python,pytorch,linux,manjaro)