首页 > 其他 > 详细

pytorch安装

时间:2019-05-15 19:45:01      阅读:139      评论:0      收藏:0      [点我收藏+]

前提条件:

  • Anaconda3 x64 (with Python 3.5/3.6)
  • Windows 64位系统(Windows 7 或 Windows Server 2008 及以上)
  • GPU版本还需要任意版本的 CUDA (包内置了CUDA 8 / 9 的部分主要二进制文件)

只需根据自己的系统键入下面的一条命令即可:(注:仅 0.3.1,以后不再更新)

# for CPU only packages
conda install -c peterjc123 pytorch-cpu

# for CUDA 8, Windows 10 or Windows Server 2016
conda install -c peterjc123 pytorch

# for CUDA 9, Windows 10 or Windows Server 2016
conda install -c peterjc123 pytorch cuda90

# for CUDA 9.1, Windows 10 or Windows Server 2016
conda install -c peterjc123 pytorch cuda91

# for CUDA 8, Windows 7 or Windows Server 2008/2012
conda install -c peterjc123 pytorch cuda80

# for CUDA 9, Windows 7 or Windows Server 2008/2012
conda install -c peterjc123 pytorch cuda90

 

pytorch安装

原文:https://www.cnblogs.com/ylHe/p/10871487.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!