首页 > 编程语言 > 详细

linux manjaro 配置 pytorch gpu 环境

时间:2019-04-21 12:36:51      阅读:246      评论:0      收藏:0      [点我收藏+]

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 才能调用独立显卡

 

 

linux manjaro 配置 pytorch gpu 环境

原文:https://www.cnblogs.com/Lin-Yi/p/10744310.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!