pythorch有适配cuda12.1的版本吗

Question

CUDA版本为12.1,但pythorch官网上给出的可选的cuda版本最高只到11.8，可以直接安装吗？

Answer 1

参考GPT和自己的思路：

目前PyTorch官网上给出的CUDA版本最高只到11.8，因此没有适配CUDA 12.1的PyTorch版本。如果你想在CUDA 12.1上使用PyTorch，建议等待官方发布适配CUDA 12.1的PyTorch版本。另外，也可以尝试自己编译最新的PyTorch源代码来适配CUDA 12.1。