linux manjaro 配置 pytorch gpu 环境

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

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

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

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

sudo pacman -S python-pytorch-cuda

就这一句就自动吧 cuda cudnn 都装好了,,再装支持gpu的pytorch就可以了

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

猜你喜欢

转载自www.cnblogs.com/Lin-Yi/p/10744310.html