安装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-pip
pip install numpy scipy matplotlib pylint
安装PyTorch
在Anaconda虚拟环境下安装PyTorch 1.4.0。可以用命令直接安装,也可以下载PyTorch的源码在本地编译安装。
安装官方包
确认开启conda的torch环境后,执行以下命令:
pip install torch torchvision
本地编译安装
- 安装本地编译所需的依赖包
conda install numpy pyyaml mkl mkl-include setuptools cmake cffi
conda install magma-cuda100 -c pytorch
- 安装Git和C++编译环境
sudo apt-get install git cmake build-essential
- 下载PyTorch源码
git clone --recursive https://github.com/pytorch/pytorch
- 编译PyTorch
cd pytorch
export CMAKE_PREFIX_PATH="/home/john/anaconda3/envs/torch"
python setup.py install
测试PyTorch
执行以下命令,打开Python交互模式
conda activate torch
python
然后测试PyTorch是否安装好
import torch
print(torch.cuda.is_available())
看输出结果是否为“True”。
看来又出现了新的问题待我把这个问题解决后再来写后续吧。