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

linux manjaro 配置 pytorch gpu 环境

2 安装支持gpu的pytorch

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

找对应自己的pyorch安装

linux manjaro 配置 pytorch gpu 环境

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

3 验证一下

用 optirun python 开python交互,,

import torch

torch.cuda.is_available()

linux manjaro 配置 pytorch gpu 环境

就成功了。

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

undefined

Original: https://www.cnblogs.com/Lin-Yi/p/10744310.html
Author: 稀里糊涂林老冷
Title: linux manjaro 配置 pytorch gpu 环境

原创文章受到原创版权保护。转载请注明出处:https://www.johngo689.com/9564/

转载文章受原作者版权保护。转载请注明原作者出处!

(0)

大家都在看

最近整理资源【免费获取】:   👉 程序员最新必读书单  | 👏 互联网各方向面试题下载 | ✌️计算机核心资源汇总