PyTorch 开发环境搭建

2021-08-26 17:38:50 浏览数 (1)

Anaconda下载及安装

简介

  • Anaconda官网:https://www.anaconda.com/
  • Anaconda是一个开源的Python发行版本,包含了180 的科学包及其依赖项,可以方便获取开发所需要的包,并且对包进行管理,其中的科学包有:condanumpy

下载

  • 网速好:官方下载:https://repo.anaconda.com/archive/
  • 网速慢:清华镜像:https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/
  • 选择相应版本点击下载

安装

  • 先卸载电脑上安装的Python,因为安装Anaconda的时候会安装相应的Python版本
  • 打开cmd命令行窗口检查安装情况,如下图所示说明未安装Python
  • 点击Anaconda安装包进行安装,其他选项按默认选择
  • 更改安装路径(注:安装路径中不要出现中文和空格)
  • 两个都勾选上
    • 第一个:将Anaconda自动添加环境变量,建议勾选,之后不用手动添加环境变量
    • 第二个:将Anaconda所带的Python3.8设置为系统默认的Python版本
  • 验证是否安装成功
    • 在终端输入Python,如果安装成功会显示Python版本号,且版本号旁边有Anaconda等字样

Warning中的内容是说环境未激活,后续操作会激活

  • 也可以输入conda list查看安装的包和其对应版本

  • 也可以打开Anaconda Powershell Prompt,显示(base)说明安装成功

创建新环境

  • 打开Anaconda Powershell Prompt
  • 语法:
代码语言:txt复制
conda create --name <env_name> <package_names>
  • 示例:
代码语言:txt复制
conda create --name py38 python=3.8
  • 过程:
  • 输入y,安装所需要的包
  • 创建完成

激活环境

  • 创建完成后,有两个选项
    • To activate this environment:输入命令激活环境
    • To deactivate an active environment:输入命令退出环境
  • 激活环境

当(base)变为环境名(py38)就说明环境已经被激活了

  • 激活后输入pip list查看环境中有哪些包

其中没有PyTorch,接下来安装PyTorch

PyTorch安装

简介

  • PyTorch是一个基于python的科学计算包,主要为使用者提供了两大功能
    • 作为Numpy的替代品, 向用户提供使用GPU的强大算力来实现神经网络的加速
    • 为用户提供了一个高灵活性,速度快的深度学习平台

conda换国内源

  • 为提高安装速度可以使用国内镜像源,提高下载速度,以下操作均在终端进行
  • 查看源
代码语言:txt复制
conda config --show-sources
  • 添加源
代码语言:txt复制
conda config --add channels 源地址
  • 推荐使用北大镜像源
代码语言:txt复制
https://mirrors.bfsu.edu.cn/anaconda/pkgs/main
https://mirrors.bfsu.edu.cn/anaconda/pkgs/r
https://mirrors.bfsu.edu.cn/anaconda/pkgs/msys2
  • 删除源
代码语言:txt复制
conda config --remove channels 源地址

安装

  • 进入官网:https://pytorch.org/
  • 下拉找到对应界面
  • 获取运行代码
    • 查看电脑显卡:可以在任务管理器->性能中查看显卡内容
    • 无英伟达显卡:Compute Platform选择cpu,并复制指令到Anaconda Powershell Prompt中运行
  • 有英伟达显卡:需要电脑上有单独的英伟达的显卡或者英伟达的显卡和集显
    • 在终端输入`nvidia-smi`查看驱动版本,如果`Driver Version`小于`400`的话需要去英伟达显卡驱动下载
  • 选择CUDA10.2,更稳定一些
  • 打开Anaconda Powershell Prompt,输入指令
  • 输入y进行安装
  • 安装完成后输入pip list,查看安装完成的包

检验安装

  • Anaconda Powershell Prompt中输入python,并输入以下代码
代码语言:txt复制
import torch
torch.cuda.is_available()
  • 如果返回True,说明安装成功

0 人点赞