win10系统pip配置gpu版本的pytorch的详细步骤

mayikengutou 23 0 PDF 2021-01-16 14:01:30

这几天在下载pytorch和相关附件的时候掉进了很多的坑,看到论坛上很多的方法是采用了linux系统或者在win10下用anacoda来完成的,但是我又懒得下载anaconda_(:3J∠)_ 所以这里写一篇避开anaconda,单纯用pip的win10下载教程吧,希望可以对其他人有帮助。 一、下载torch和torchvision 1.在cmd中用pip下载 pip install torch torchvision 由于下载速度太慢,该方法一般会失败。 2.在官网下载wheel文件 登录pytorch官网 根据需要选择相应的版本,这里选择的是cuda10.1的版本(None是cpu版本)

用户评论
请输入评论内容
评分:
暂无评论