zoukankan      html  css  js  c++  java
  • 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 才能调用独立显卡

    
    
  • 相关阅读:
    SQL数据库常用命令
    软件测试基础学习
    单链表面试题集合
    常见算法排序,冒泡排序,快排,堆排,归并排序
    CSS学习笔记(2)
    CSS学习笔记(1)
    sublime快捷键
    Sublime Text 中文输入法无法跟随怎么办
    网站收集
    Centos7安装Jenkins
  • 原文地址:https://www.cnblogs.com/Lin-Yi/p/10744310.html
Copyright © 2011-2022 走看看