正确版本的pytorch是1.2.0，cuda是10.0，但是我服务器只支持cuda11.0以上版本，我想通过修改代码来适应我的服务器cuda版本，应该修改哪里？...
·
如果你的服务器只支持 CUDA 11.0 或以上版本，你需要升级你的 PyTorch 版本以适应此 CUDA 版本。
你可以在 PyTorch 的官方网站上下载与 CUDA 11.0 兼容的 PyTorch 版本。下载完成后，你需要将代码中的所有 PyTorch 相关依赖更新为新版本。
如果你在代码中指定了 CUDA 版本，你需要将其更新为 11.0 或更高版本。
除此之外，你需要确保你的服务器已经安装了 CUDA 11.0，并且你的代码可以正确地使用 CUDA 11.0。你可以通过在服务器上运行简单的 CUDA 代码来测试 CUDA 是否正常工作。
总之，要适应服务器的 CUDA 版
欢迎来到FlagOS开发社区，这里是一个汇聚了AI开发者、数据科学家、机器学习爱好者以及业界专家的活力平台。我们致力于成为业内领先的Triton技术交流与应用分享的殿堂，为推动人工智能技术的普及与深化应用贡献力量。
更多推荐
0
0- 0
已为社区贡献1条内容
所有评论(0)