下载GPU版本的pytorch

作者：2023面试高手 | 2024-04-05 01:39:12

踩

下载gpu版本的pytorch

1.下载cuda和cudnn

cuda下载网址：（version：11.8）
https://developer.nvidia.com/cuda-11-8-0-download-archive
cudnn下载网址：
https://developer.nvidia.com/cudnn

二者版本一定要对应！！

cuda按照安装导航一步一步点同意，cudnn下载解压后放到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8目录下：
在这里插入图片描述

然后配置环境如图：
在这里插入图片描述

下载网址：
https://pytorch.org/get-started/locally/
在这里插入图片描述

我这样下载以后出现了这样的报错：
在这里插入图片描述
根据这位

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/2023面试高手/article/detail/362477