2021-10-12 PyTorch简介和安装

PyTorch简介和安装

简介

  • python深度学习库
  • 简介优雅且⾼效快速
  • 可以更好地调试代码
    可逐⾏执⾏,类似调试NumPy—可轻松访问所有对象,且使⽤打印语句查看失败位置

安装

Anaconda:实现python包管理和环境管理

  • 环境管理:每个环境下是独⽴的运⾏空间,包的依赖(低级包给⾼级包提供依赖)不会 产⽣⽭盾
  • 命令
    • 查看虚拟环境:

    conda info —env

    • 创建虚拟环境:

    conda create -n 名称 python==版本

    • 删除虚拟环境:

    conda remove -n 名称 -all

    • 激活环境:

    conda activate 名称

  • conda换源(换清华源)

    • 移除:

    conda config —remove -key channels

    • 查看:

    conda config —show channels

    • 修改:

    修改.condrac⽂件:
    vim .condrac
    输⼊i进⼊编辑模式,粘贴内容
    channels:
    -defaults
    show_channel_urls:true
    default_channels:
    -https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
    -https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/r
    -https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/msys2
    custom_channels:
    conda-forge: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
    msys2: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
    bioconda: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
    menpo:https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
    按 ESC 退出编辑模式,输⼊ :wq保存并退出

  • pip换源

    • ⽤户⽬录下新建pip⽂件夹:

    mkdir ~/.pip

    • ⽂件内新建pip.conf:

    vim ~/.pip/pip.conf

    • 添加以下内容:

    [global]
    index-url = http://pypi.douban.com/simple
    [install]
    use-mirrors =true
    mirrors = http://pypi.douban.com/simple/
    trusted-host = pypi.douban.com

PyTorch安装:官⽹根据需要选择安装版本和CUDA配置

  • 复制命令,粘贴一键安装(建议使用conda安装)
  • stable稳定版本,preview先⾏版本
  • 建议使⽤anaconda管理

检查安装是否完成

import torch
torch.cuda.is_available()

CPU将返回False,GPU返回true,否则报错

相关资源

上一篇:ExtJs 备忘录(6)—— GirdPanl表格(二) [ 搜索分页 ]


下一篇:MultipartFile实现文件的上传和下载