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)

大家都在看

发表回复

登录后才能评论
免费咨询
免费咨询
扫码关注
扫码关注
联系站长

站长Johngo!

大数据和算法重度研究者!

持续产出大数据、算法、LeetCode干货,以及业界好资源!

2022012703491714

微信来撩,免费咨询:xiaozhu_tec

分享本页
返回顶部