Windows10+GPU版 pytorch 安装
用官网pip的命令太慢,换国内的源也不行,因为国内的源会默认给你安装cpu版本。在安装成功之前经历过很多失败,要么是安装成cpu版本;要么是安装的版本和cuda不符合;要么就是pytorch下载到一半,网速不好,直接安装失败。
这里是先将pytorch下载到自己的电脑上,然后再进行安装。
一、首先确保已经安装cuda和cudnn
这里是在确保已经安装好cuda和cudnn的情况下进行的,cuda和cudnn的相关安装教程可以自己搜一下。
这里我安装的cuda版本为10.0
[外链图片转存失败,源站可能有防盗
原创
2021-08-01 20:13:14 ·
313 阅读 ·
0 评论