1. 安装 cuda 环境

1
sudo pacman -S python-pytorch-cuda

2. 安装 pytorch 环境

1
pip install torch torchvision

也可以用其他命令比如 conda 等,具体可以参照 pytorch 官网

3. 测试

1
python

进入到 python 命令行后运行:

1
2
import torch
torch.cuda.is_available()

如果返回结果是 True,则说明环境已经搭建好;如果返回是 False,则说明环境还有问题。如果上述安装都没有问题,那么可能和笔者一样,是 Manjaro 当前启用了开源的 Nouveau 显卡驱动,需要将其禁用,然后再安装最新的 Nvidia 闭源驱动(详见下文)。

4. 问题 & 解决

  • 问题: 如果上述安装都没有问题,输入以下命令测试当前系统 Nvidia 闭源驱动是否工作良好:
1
nvidia-smi

如果当前 Manjaro 启用了开源的 Nouveau 显卡驱动,或者启用的不是最新的闭源驱动,都会导致该命令输出报错信息:NVIDIA-SMI has failed because it couldn't communicate with the NVIDIA driver. Make sure that the latest NVIDIA driver is installed and running.

  • 解决:到 Nvidia 官网下载匹配自己电脑显卡版本的最新 Linux 闭源驱动(NVIDIA-Linux-x86_64-xxx.run),然后终端安装执行。如果是因为当前 Manjaro 启用了开源的 Nouveau 显卡驱动,终端会给予提示,只要跟着终端提供的方法走,基本就能成功禁用 Nouveau 开源驱动,然后安装上最新 Nvidia 闭源驱动(Nvidia 闭源驱动 .run 安装包会提供生成禁用 Nouveau 开源驱动的配置文件,然后只要重启系统后重新执行 .run 文件安装 Nvidia 闭源驱动一般即可成功)。