maybegreat的博客

原创 博客摘录「 RTX3090(4090) 与pytorch版本对应关系」2023年10月25日

RTX3090 至少需要cuda 11.1 版本，才能够驱动该设备, 故我们可以安装cuda11.1 以上版本所以在想要安装的 pytorch 版本中，要安装的pytorch-cuda --version。当前机器上安装的 cuda --version。大于cuda11.1。