虚拟环境无法隔离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

魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。

更多推荐