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 | import torch |
如果返回结果是 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 闭源驱动一般即可成功)。
本博客所有文章除特别声明外,均采用 CC BY-NC-SA 4.0 许可协议。转载请注明来自 お前はどこまで見えている!
评论
WalineTwikoo