本文共 1295 字，大约阅读时间需要 4 分钟。

安装Anaconda

在官网下载Anaconda安装包：https://www.anaconda.com/distribution/#linux，比如https://repo.anaconda.com/archive/Anaconda3-2019.10-Linux-x86_64.sh。然后执行以下命令安装Anaconda：

chmod +x Anaconda3-2019.10-Linux-x86_64.sh./Anaconda3-2019.10-Linux-x86_64.sh

安装过程中提示安装位置时，建议选择安装在个人目录下，比如~/anaconda3。在提示是否允许修改.bashrc文件时，建议输入“yes”。

重新打开终端，正常情况下，在用户名前会显示“(base)”字样，说明已经进入conda的默认环境。如果没有该字样的话，可以执行conda init，以激活conda环境。

用以下命令建立一个名为”torch”的Python 3.7环境：

conda create -n torch python=3.7

执行以下命令开启该环境：

conda activate torch

(torch) 表示进入了该环境

如果需要关闭，执行以下命令关闭该环境：

conda deactivate

可以顺便把常用的库安装好：

sudo apt-get install python3-pippip install numpy scipy matplotlib pylint

安装PyTorch

在Anaconda虚拟环境下安装PyTorch 1.4.0。可以用命令直接安装，也可以下载PyTorch的源码在本地编译安装。

安装官方包

确认开启conda的torch环境后，执行以下命令：

pip install torch torchvision

本地编译安装

1. 安装本地编译所需的依赖包

conda install numpy pyyaml mkl mkl-include setuptools cmake cffi

conda install magma-cuda100 -c pytorch

2. 安装Git和C++编译环境

sudo apt-get install git cmake build-essential

3. 下载PyTorch源码

git clone --recursive https://github.com/pytorch/pytorch

4. 编译PyTorch

cd pytorchexport CMAKE_PREFIX_PATH="/home/john/anaconda3/envs/torch"python setup.py install

测试PyTorch

执行以下命令，打开Python交互模式

conda activate torchpython

然后测试PyTorch是否安装好

import torchprint(torch.cuda.is_available())

看输出结果是否为“True”。

看来又出现了新的问题待我把这个问题解决后再来写后续吧。

转载地址：http://uxtzi.baihongyu.com/