linux manjaro 配置 pytorch gpu 环境

manjaro目前中国资料偏少，踩了很多坑。

安装gpu版本就这么几个步骤 1 安装英伟达的驱动cuda  2 安装 cudnn   3 安装支持gpu的pytorch 或者其他的运算框架

 

manjaro中文资料很少，，，一步一步先装cuda 再装cudnn  好麻烦，各种错误

 

实际上就这一句就解决了：

1 装 cuda 和 cudnn

sudo pacman -S python-pytorch-cuda

就这一句就自动吧 cuda cudnn 都装好了，

这句话执行中间会失败很多次，，多几次重新执行，它把所有包下载好了 就能成功了。


然后使用命令查看一下自己的cuda 是什么版本
optirun nvidia-smi

 

 

2 安装支持gpu的pytorch

去官方网站 https://pytorch.org/

找对应自己的pyorch安装

 

我的cuda 是10.1  但是我发现网站上pytorch 支持cuda的只到10， 所以我就安装了cuda10的pytorch。。。。后来也成功了

 

3  验证一下

用  optirun  python 开python交互，，

import torch

torch.cuda.is_available()

 

就成功了。

 

据说manjaro系统不会自动切换独立显卡，，所以运行时候要加上optirun 才能调用独立显卡