在Ubuntu上编译安装PyTorch步骤如下:
sudo apt update
,sudo apt install -y build-essential cmake git python3-dev python3-pip libopenblas-dev liblapack-dev libjpeg-dev libpng-dev
等。python3 -m venv pytorch_env
,source pytorch_env/bin/activate
。git clone --recursive https://github.com/pytorch/pytorch
,然后进入仓库目录cd pytorch
。export USE_CUDA=1
等。python setup.py install
进行编译安装,此过程可能耗时较长。import torch
,print(torch.__version__)
,若输出版本号则安装成功。若要安装GPU版本,需先安装NVIDIA驱动、CUDA和cuDNN,再在编译时设置USE_CUDA=1
等环境变量。