Manjaro20.1下搭建Pytorch环境

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 闭源驱动一般即可成功）。