安装前注意自己的python版本,我已安装好python3.6,之后都在3.6下进行。
torch:
在官网选择相应版本。Start Locally | PyTorch
pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cpu
tensorflow
在网站上选择相应的版本
export TF_BINARY_URL=https://storage.googleapis.com/tensorflow/linux/cpu/tensorflow-1.1.0-cp36-cp36m-linux_x86_64.whl
pip3 install $TF_BINARY_URL
由于该网站版本较低,需要升级版本的可以通过以下命令升级。XXX表示将升级到的版本。
pip install --upgrade tensorflow==X.X.X