美文网首页
GPU版的pytorch安装

GPU版的pytorch安装

作者: Bystander_1996 | 来源:发表于2020-01-20 15:46 被阅读0次

    一:Anconda方式安装

    1.1 安装anaconda

    安装anaconda或者miniconda
    进入清华镜像站下载自己需要的anaconda版本
    anaconda下载地址

    image.png
    执行安装命令: bash + 下载的安装包
    可能的出错是没有安装bzip2 安装一下sudo apt-get install bzip2
    安装后如果找不到命令一般是由于环境变量未导入,或者环境变量导入了未更新。输入命令打开配置文件vim ~/.bashrc
    在最后添加语句export PATH=/home/XXX/anaconda3/bin:$PATH
    更新配置文件: source ~/.bashrc

    1.2 anaconda更换清华镜像源

    conda config --add channels [https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/](https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/)
    conda config --set show_channel_urls yes       
    //输入这两行命令即可换源成功。
    

    anaconda常用命令

    //更新所有库
    conda update --all
    //更新 conda 自身
    conda update conda
    //查看已经安装的包:
    pip list 或者 conda list
    //安装和更新:
    pip install requests
    pip install requests –upgrade
    conda install requests
    conda update requests
    

    1.2 创建conda虚拟环境并进入:

    conda create -n env_name pyhton=3.6  //创建环境 
    source activate env_name  //激活并进入环境
    

    1.3 安装pytorch

    进入pytorch官网选择安装的命令 附:pytorch官网

    conda安装方式
    根据自己的系统,python语言版本等如实选择就可以得到安装命令,执行安装操作即可

    二:pip安装

    2.1 配置pip

    更换pip的镜像站
    修改 ~/.pip/pip.conf (没有就创建一个),修改内容如下

    [global]
     index-url = https://pypi.tuna.tsinghua.edu.cn/simple
    

    临时换源
    在使用pip的时候加参数 -i https://pypi.tuna.tsinghua.edu.cn/simple
    例:pip install gevent -i https://pypi.tuna.tsinghua.edu.cn/simple\

    更新pip
    python3 -m pip install --upgrade pip

    2.2 安装pytorch

    进入pytorch官网选择安装的命令 附:pytorch官网

    pip安装方式

    根据自己的系统,python语言版本等如实选择就可以得到安装命令,执行安装操作即可

    三:查看是否安装成功

    打开python

    import torch
    print(torch.cuda.is_available())
    
    image.png
    torch.cuda.is_available()的结果是True 证明是可以使用的,为了保险又用数据试了一下,也是没有问题的

    参考:https://www.jianshu.com/p/e298b9d3afae

    相关文章

      网友评论

          本文标题:GPU版的pytorch安装

          本文链接:https://www.haomeiwen.com/subject/qtmjzctx.html