问安装11.2CUDA的PyTorch GPU
EN

Stack Overflow用户

提问于 2021-12-17 16:30:03

回答 1查看 1.6K关注 0票数 0

如果我的数据自动化系统版本为11.2，而PyTorch版本为8.1.0，如何安装CudNN？

https://pytorch.org/get-started/locally/的官方文件建议只有CUDA 10.2和11.3版本。

我是否应该用下一个命令用10.2版本安装它呢？

conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch

将来不会有任何不兼容的版本问题吗？

发布于 2022-09-26 08:04:41

据我所知，tell1没有为CUDA 11.2提供预编译库。你必须自己编译它。要了解这一点，请阅读PyTorch Github's README的这一节。

他说:我这么认为是因为在PyTorch论坛上提出了类似的问题，其中一位主持人说了同样的话-- https://discuss.pytorch.org/t/want-to-install-pytorch-for-custom-cuda-version-cuda-11-2/141159/2。

编辑:您可以安装为CUDA v11.1编译的PyTorch。在我的例子中，它似乎没有任何问题。

票数 0

页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持

原文链接：

https://stackoverflow.com/questions/70396187

复制

相似问题

问安装11.2CUDA的PyTorch GPUEN