1. 安裝 cuda 環境
sudo pacman -S python-pytorch-cuda
2. 安裝 pytorch 環境
pip install torch torchvision
也可以用其他命令比如 conda 等，具體可以參照 pytorch 官網。
3. 測試
python
進入到 python 命令行後運行：
import torch
torch.cuda.is_available()
如果返回結果是 True，則說明環境已經搭建好；如果返回是 False，則說明環境還有問題。如果上述安裝都沒有問題，那麼可能和筆者一樣，是 Manjaro 當前啓用了開源的 Nouveau 顯卡驅動，需要將其禁用，然後再安裝最新的 Nvidia 閉源驅動（詳見下文）。
4. 問題 & 解決
- 問題： 如果上述安裝都沒有問題，輸入以下命令測試當前系統 Nvidia 閉源驅動是否工作良好：
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 閉源驅動一般即可成功）。