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