1. 首页
  2. 数据库
  3. 其它
  4. win10系统pip配置gpu版本的pytorch的详细步骤

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

上传者: 2021-01-16 14:11:30上传 PDF文件 1.01MB 热度 24次
这几天在下载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版本)
下载地址
用户评论