windows 10 anaconda python 3.7 安装 pytorch-gpu

win 10 anaconda python 3.7 安装 pytorch

2019-6-1:清华更新源已经关闭了

先添加清华源:


安装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

相关推荐
©️2020 CSDN 皮肤主题: 编程工作室 设计师:CSDN官方博客 返回首页