cuda版本12.1可以装cuda11.4版本的pytorch吗
时间: 2024-05-31 19:13:54 浏览: 321
机器学习,pytorch英伟达显卡驱动,cndann,适用于cuda12.1以上
一般来说,CUDA 12.1和CUDA 11.4是不兼容的,因此无法直接安装CUDA 11.4版本的PyTorch。但是,你可以考虑使用虚拟环境来安装不同版本的PyTorch,这样你就可以在同一台计算机上安装多个版本的PyTorch,并在需要时切换到不同的环境。具体来说,你可以使用conda或者virtualenv创建虚拟环境,并在每个环境中安装特定版本的PyTorch和它所需要的CUDA版本。这样,你就可以在不同的虚拟环境中使用不同版本的PyTorch了。
阅读全文