Ubuntu18.04安装cpu版pytorch环境

环境说明

• 操作系统：Ubuntu 18.04 LTS
• anaconda版本：Anaconda3-2020.11-Linux-x86_64.sh
• anaconda中创建的python：Python 3.8.8

anaconda安装

• 访问https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/，自行选择anaconda版本，下载存放至/home/fmr/Downloads
• 解压安装并配置环境变量

  cd ~/Downloads/
  chmod +x Anaconda3-2020.11-Linux-x86_64.sh
  ./Anaconda3-2020.11-Linux-x86_64.sh
  # 安装过程中会提示选择Yes No，一般不需要改动，回车即可
  #默认anaconda安装在用户的home目录下
  echo 'export PATH="/home/fmr/anaconda3/bin:$PATH"' >> ~/.bashrc
  source ~/.bashrc
  

• 验证是否安装成功

  conda info -e
  # conda environments:
  #
  base                     /home/fmr/anaconda3
  #若出现类似上述信息，说明安装成功
  

修改anaconda配置

• 激活anaconda环境

  conda init bash
  ctrl + D #退出当前登录的shell
  #重新登录，若anaconda正常运行，shell提示符应该与以下内容类似
  (base) fmr@fmr-virtual-machine:~/Downloads$
  

• 配置anaconda清华源

  conda config --set show_channel_urls yes 
  #上述指令会生成文件/home/fmr/.condarc
  
  vi /home/fmr/.condarc
  #用以下内容替换/home/fmr/.condarc中原本的内容
  
  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
    pytorch: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
    simpleitk: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  
  conda clean -i #清除索引，保证使用的是镜像站提供的索引
  

pytorch（cpu）安装

• 创建python虚拟环境

  conda create -n pytorch-cpu python=3.8 # 创建python3.8虚拟环境
  conda activate pytorch-cpu # 进入名为pytorch-cpu的虚拟环境
  

• 访问https://pytorch.org/get-started/locally/，查看合适的下载指令

  image.png

• 使用查询到的下载指令，安装pytorch

  conda install pytorch torchvision torchaudio cpuonly -c pytorchd
  

• 验证安装是否成功

  conda activate pytorch-cpu #进入pytorch-cpu虚拟环境
  python #进入python 交互环境
  
  #若与下列内容类似，说明pytorch安装成功
  Python 3.8.8 (default, Apr 13 2021, 19:58:26) 
  [GCC 7.3.0] :: Anaconda, Inc. on linux
  Type "help", "copyright", "credits" or "license" for more information.
  >>> import torch 
  >>> print(torch.cuda.is_available())
  False
  >>> 
  

设置vscode远程连接

• 开启Ubuntu ssh服务

  sudo apt-get install ssh
  sudo vi /etc/ssh/sshd_config
  
  #按照下列内容修改/etc/ssh/sshd_config文件中相对应的内容
    Port 22
    #AddressFamily any
    ListenAddress 0.0.0.0
    #ListenAddress ::
  
    #LoginGraceTime 2m
    PermitRootLogin yes
    #StrictModes yes
    #MaxAuthTries 6
    #MaxSessions 10
  
  service ssh start #开启Ubuntu ssh
  

• 设置vscode远程连接

  1. 安装 remote ssh插件

     
     
     image.png

  2. 点击下图“1”所示位置，而后点击“2”处的加号，新建ssh连接

     
     
     image.png

  3. 在画圈位置输入ssh指令

     
     
     image.png
  4. 按回车，会出现选择保存位置的下拉框，此处选择第一个：保存至当前用户下

  5. 点击画圈位置的加号，连接远端服务器

     
     
     image.png
  6. 根据提示，输入与登录用户对应的密码，并选择一个文件夹打开

  7. 以后只需要点击下图画圈部分，就可以直接打开一个新窗口，并自动连接至该文件夹目录下。

     
     
     image.png

补充

• 如果是新安装的Ubuntu虚拟机，使用vi、vim编辑文件的时候，可能会出现，按方向键无法移动光标而是输出字母，BackSpace键也不能删除字符
• 这是因为Ubuntu预装的是vim-tiny，解决方法如下

  sudo apt-get remove vim-common
  sudo apt-get install vim