先安装好cuda和cudnn,这是cuda和cudnn 11.1的版本,下面百度云有安装包和安装说明的网站。
conda安装GPU版本的pytorch
最新推荐文章于 2024-10-10 16:18:12 发布
本文介绍了如何在Windows 10上使用Anaconda安装CUDA 11.1和CUDNN,以及详细步骤来安装GPU支持的PyTorch 1.9.1和torchvision 0.10.1。首先从指定链接下载CUDA和CUDNN安装包,然后按照提供的安装指南进行安装。接着,通过conda创建环境并激活,使用pip在环境中安装PyTorch和torchvision的特定版本。最后,通过设置pip的全局索引URL到阿里云镜像,确保后续包的安装速度。

最低0.47元/天 解锁文章
6万+

被折叠的 条评论
为什么被折叠?



