ubuntu20.04安装pytorch运行环境(附安装问题)

python3.7安装

  1. 卸载默认版本
    Ubuntu默认的python版本一般为python 2.x,因此需要手动安装自己需要的版本,并且将python的指向更新为自己的版本。如果已经使用apt-get install python安装了python2.x的版本,可以先卸载
sudo apt-get remove python2.7
sudo apt-get remove --auto-remove python2.7
sudo apt-get purge python2.7
sudo apt-get purge --auto-remove python2.7

  1. 下载并解压安装包

下载安装包

$ wget https://www.python.org/ftp/python/3.7.1/Python-3.7.1.tgz
  1. 解压安装包
    $ tar -zxvf Python-3.7.1.tgz

  2. 进入到解压后的Python-3.7.1目录下
    $ cd Python-3.7.1

  3. 创建安装目录
    在/usr/local/目录底下创建一个安装目录

$ sudo mkdir -p /usr/local/python3

  1. 编译安装

顺序执行下面三个命令

$ ./configure --prefix=/usr/local/python3  --enable-optimizations
$ make
$ sudo make install

如果执行make install时出现以下错误

Python3: ImportError: No module named '_ctypes' when using Value from module multiprocessing

进一步安装依赖后再重新make install

sudo apt-get install libffi-dev
  1. 删除python和pip的软链接

删除python软链接

rm -rf /usr/bin/python

删除pip软链接

 rm -rf /usr/bin/pip
  1. 添加python和pip的软链接

添加python软链接

ln -s /usr/local/python3/bin/python3.7 /usr/bin/python

添加 pip3 的软链接

ln -s /usr/local/python3/bin/pip3.7 /usr/bin/pip
  1. 确认安装成功
    查看python版本
    在这里插入图片描述
    查看pip版本在这里插入图片描述

安装Anaconda

Anaconda是一个开源的Python发行版本,包含了conda、Python等180多个科学包及其依赖项,可以用于包管理器和环境管理。比如A项目中用了Python2,而B项目使用Python3,而同时安装两个Python版本可能会造成许多混乱和错误,这是使用Anaconda就可以为不同的项目建立不同的运行环境。

  1. 下载Anaconda安装包:

wget https://repo.continuum.io/archive/Anaconda3-5.3.1-Linux-x86_64.sh

除了上面的下载方式,也可以使用清华的镜像地址

wget https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/Anaconda3-5.3.1-Linux-x86_64.sh

  1. 进入安装包所在目录,使用bash指令进行anaconda安装:

bash Anaconda3-5.3.1-Linux-x86_64.sh

安装过程会出现是否阅读注册信息、添加环境变量、安装VSCode等提示信息,全部按yes即可。
如果安装过程出现 “Unable to install python extensions for VSCode!”错误:
ubuntu20.04安装pytorch运行环境(附安装问题)_第1张图片
原因是extensions没有权限,更改目录权限就可以了

sudo chown -R 你的用户名 ~/.vscode/extensions

参考:https://www.jianshu.com/p/5d0ac9165111

  1. 查看安装结果:

conda info

若出现conda: command not found的错误,可能是因为~/.bashrc文件没有配置好

vim ~/.bashrc

在最后一行加上

export PATH=$PATH:/home/用户名/anaconda3/bin
在这里插入图片描述

安装pytorch

  1. conda创建虚拟环境

conda create -n pytorch_gpu python=3.7

  1. 安装pytorch

2.1 Conda安装pytorch
conda install pytorch torchvision torchaudio cudatoolkit=11.0

2.2 Pip安装pytorch(下载安装相对较快,推荐)
pip install torch1.7.1+cu110 torchvision0.8.2+cu110 torchaudio===0.7.2 -f https://download.pytorch.org/whl/torch_stable.html

  1. 测试安装成功ubuntu20.04安装pytorch运行环境(附安装问题)_第2张图片

开始愉快学习pytorch啦!
https://pytorch.org/get-started/locally/

你可能感兴趣的:(机器学习,pytorch,python,ubuntu)