Ubuntu18.04安装cpu版pytorch环境

2021-04-22 本文已影响0人 funOfFan

环境说明

操作系统：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
```

Ubuntu18.04安装cpu版pytorch环境

猜你喜欢

热点阅读