码迷,mamicode.com
首页 > Windows程序 > 详细

Linux/Windows GPU/CPU版 pytorch安装

时间:2018-06-27 23:56:10      阅读:745      评论:0      收藏:0      [点我收藏+]

标签:老版本   ext   lin   9.png   info   pip   win   https   .com   

Windows上安装:

最新0.4.0版本:

在Pytorch官网 https://pytorch.org/ 上选择对应版本安装即可,conda安装的比较慢,建议选择pip安装的(虽然还是很慢),当然能找到不错的镜像也是极好的。安装CPU版的在CUDA处选None。

0.3.0等以前的老版本:

建议参考知乎https://www.zhihu.com/question/67209417,用里面的镜像会快一些。

Linux上安装:

直接pip install http://download.pytorch.org/whl/cu80/torch-0.2.0.post3-cp36-cp36m-manylinux1_x86_64.whl

把后面直接换成想要的版本即可,具体可参考官网。

GPU安装后的测试:

技术分享图片

一些坑:

GPU 1080 以下的版本尽量安装0.3.0的,安装0.4.0的在Windows上会不支持。

 

Linux/Windows GPU/CPU版 pytorch安装

标签:老版本   ext   lin   9.png   info   pip   win   https   .com   

原文地址:https://www.cnblogs.com/sbj123456789/p/9236455.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!