今天一天就安装了一个pytorch,整个过程都是遇错找解,所以记录一下大致流程,方便日后再次安装时避坑。
在anaconda prompt下先安装cuda,可以百度匹配自己电脑显卡的cuda版本,之后再安装cudnn。这样就可以安装pytorch了,应注意的是pytorch与cuda版本应匹配。显卡决定了我的cuda版本,cuda决定了我的pytorch版本。
windows下安装pytorch
最新推荐文章于 2024-07-27 17:33:32 发布