虚拟环境无法隔离cuda Runtime(cuda 12.1 、cuda 11.8)

但是可以在系统上分别下载cuda12.1和cuda11.8。

系统使用的cuda版本由软链接 /usr/local/cuda决定。

分别创建虚拟环境venv_11.8,venv_12.1。

在两个环境中,分别安装cuda11.8版的torch和cuda12.1版的torch。

之后激活环境,正常使用即可。因为安装时指定了cuda版本,使用torch时会默认找系统的11.8/12.1

Logo

欢迎来到FlagOS开发社区,这里是一个汇聚了AI开发者、数据科学家、机器学习爱好者以及业界专家的活力平台。我们致力于成为业内领先的Triton技术交流与应用分享的殿堂,为推动人工智能技术的普及与深化应用贡献力量。

更多推荐