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

展开阅读全文

没有更多推荐了,返回首页

©️2019 CSDN 皮肤主题: 编程工作室 设计师: CSDN官方博客
应支付0元
点击重新获取
扫码支付

支付成功即可阅读