Java自学者论坛

 找回密码
 立即注册

手机号码,快捷登录

恭喜Java自学者论坛(https://www.javazxz.com)已经为数万Java学习者服务超过8年了!积累会员资料超过10000G+
成为本站VIP会员,下载本站10000G+会员资源,会员资料板块,购买链接:点击进入购买VIP会员

JAVA高级面试进阶训练营视频教程

Java架构师系统进阶VIP课程

分布式高可用全栈开发微服务教程Go语言视频零基础入门到精通Java架构师3期(课件+源码)
Java开发全终端实战租房项目视频教程SpringBoot2.X入门到高级使用教程大数据培训第六期全套视频教程深度学习(CNN RNN GAN)算法原理Java亿级流量电商系统视频教程
互联网架构师视频教程年薪50万Spark2.0从入门到精通年薪50万!人工智能学习路线教程年薪50万大数据入门到精通学习路线年薪50万机器学习入门到精通教程
仿小米商城类app和小程序视频教程深度学习数据分析基础到实战最新黑马javaEE2.1就业课程从 0到JVM实战高手教程MySQL入门到精通教程
查看: 818|回复: 0

Linux安装pytorch的具体过程以及其中出现问题的解决办法

[复制链接]
  • TA的每日心情
    奋斗
    2024-11-24 15:47
  • 签到天数: 804 天

    [LV.10]以坛为家III

    2053

    主题

    2111

    帖子

    72万

    积分

    管理员

    Rank: 9Rank: 9Rank: 9

    积分
    726782
    发表于 2021-5-17 12:51:54 | 显示全部楼层 |阅读模式

    1.安装Anaconda

      安装步骤参考了官网的说明:https://docs.anaconda.com/anaconda/install/linux.html

      具体步骤如下:

         首先,在官网下载地址 https://www.anaconda.com/download/下载linux版本,这里选用python 3.6版本的anaconda.

         然后, 将下载好的Anaconda3-4.4.0-Linux-x86_64.sh放到/usr/tiny目录下,并进入该目录

         在当前目录下用bash命令安装anaconda

     

    root bash    Anaconda3-4.4.0-Linux-x86_64.sh 
       安装完毕后,输入如下命令使得配置生效

    source ~/.bashrc

      为验证安装正确性,可以通过如下命令进行测试

    conda list
    

    2.用conda安装/更新pytorch的依赖库

    首先,激活Anaconda配置文件

    source ~/.bashrc
    

    使用conda install命令可以方便快捷得安装或者更新所需的内容

    conda install numpy pyyaml mkl setuptools cmake gcc cffi
    

     

    3.安装pytorch

    本文对pytorch官网给出的三种安装方式分别进行了测试,结果如下:

    首先,使用conda进行安装,命令如下,其中cuda80可以根据cuba版本调整为cuda75或者直接省略

     

    conda install pytorch torchvision cuda80 -c soumith
    

    因为下载速度只有几KB,忍无可忍直接终止了进程。

    然后直接从官网下载了source文件

    git clone --recursive https://github.com/pytorch/pytorch

    然后用setup.py进行安装

    python setup.py install
    

    安装过程显示报错,缺失python.h文件,于是把python-dev和python3-dev一起装了一遍 

    sudo apt-get install python-dev 
    sudo apt-get install python3-dev 

    再次运行 

    python setup.py install
    

    显示把torch安装到了python2.7的库文件夹里,本来想删掉python-dev再安装一遍pytorch,没有找到相应的办法,遂考虑通过pip进行安装

    通过pip进行pytorch安装的过程如下:

    pip install http://download.pytorch.org/whl/cu75/torch-0.2.0.post3-cp36-cp36m-manylinux1_x86_64.whl  
    pip install torchvision    
    

    下载速度在几MB左右,安装过程十分迅速,且没有报错。

    安装完毕之后,激活Anaconda配置文件

    source ~/.bashrc
     
    

    进入python命令行模式:

    python
    

    测试安装是否正确:

    from __future__ import print_function
    import torch
    a = torch.ones(5)
    print(a)
    哎...今天够累的,签到来了1...
    回复

    使用道具 举报

    您需要登录后才可以回帖 登录 | 立即注册

    本版积分规则

    QQ|手机版|小黑屋|Java自学者论坛 ( 声明:本站文章及资料整理自互联网,用于Java自学者交流学习使用,对资料版权不负任何法律责任,若有侵权请及时联系客服屏蔽删除 )

    GMT+8, 2025-2-3 10:19 , Processed in 0.061011 second(s), 28 queries .

    Powered by Discuz! X3.4

    Copyright © 2001-2021, Tencent Cloud.

    快速回复 返回顶部 返回列表