windows anaconda python 3.7 安装 pytorch-gpu

windows anaconda python 3.7 安装 pytorch

先添加清华源：

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --set show_channel_urls yes

---

安装GPU版pytorch

conda install pytorch torchvision cudatoolkit=10.0 -c pytorch

注意：cudatoolkit的版本需要适合你的显卡

---

安装CPU版pytorch

conda install pytorch-cpu torchvision-cpu -c pytorch

---

keras-gpu和tensorflow-gpu的安装参考这篇文章：windows anaconda python 3.7 安装keras-gpu tensorflow-gpu