Linux如何安装并查看Pytorch、TensorFlow

安装并查看Pytorch、TensorFlow

1.Pytorch

1.1Linux环境下安装Pytorch

​使用conda安装指定版本

conda install pytorch=1.3.1 -c soumith

使用pip安装指定版本

pip install pytorch=1.3.1

1.2Linux环境下卸载Pytorch

pip uninstall torch #pip
conda uninstall pytorch #conda

1.3Linux环境下更新pytorch

# 列举pip当前可以更新的所有安装包
pip list --outdated --format=legacy

# 更新pytorch和torchvision安装包
pip install --upgrade pytorch torchvision

1.4 Linux环境下查看pytorch版本

import torch
print(torch.__version__)

2.TensorFlow

2.1Conda 安装TensorFlow

2.1.1直接安装制定版本

conda install tensorflow-gpu==1.10

2.1.2查看tensorflow各个版本并选择其中某个版本

anaconda search -t conda tensorflow

找到自己安装环境对应的最新TensorFlow后（可以在终端搜anaconda，定位到那一行），然后查看指定tensorflow版本信息

anaconda show anaconda/tensorflow

最下面提供一个下载地址，例如使用下面命令就可安装1.8.0版本tensorflow

conda install --channel https://conda.anaconda.org/anaconda tensorflow=1.8.0

2.2 Linux环境下查看TensorFlow版本

import tensorflow
print(tensorflow.__version__)