cuda11.8和pytorch版本对应
时间: 2024-04-06 13:24:42 浏览: 1069
浅谈pytorch、cuda、python的版本对齐问题
5星 · 资源好评率100%
根据引用和引用的内容,可以得知目前PyTorch官网最高支持的CUDA版本是11.8,但是有些滞后。如果你的CUDA版本是12.1.66,那么需要验证一下支持CUDA11.8的PyTorch是否和CUDA12.1.66是兼容的。同时,引用提供了一个基于CUDA11.7和PyTorch的CNN简单例子,但是需要注意的是,CUDA11.7和CUDA11.8并不完全相同,因此需要根据实际情况进行选择。总之,为了保证CUDA和PyTorch的兼容性,建议在安装PyTorch之前先了解清楚官网所支持的CUDA版本,并根据自己的实际情况进行选择。
阅读全文