选择要在Singularity 3.6容器中使用的特定CUDA版本

2024-04-26 03:20:05 发布

您现在位置:Python中文网/ 问答频道 /正文

我正在通过Singularity 3.6开发运行tensorflow容器的POC。我有一个要求,用户可能需要选择一个旧版本的CUDA,这是在一个定制的位置,而不是系统安装的一个

我阅读了关于GPGPU支持的3.6文档,-nv标志自动执行所有功能,我看不到任何可以指定使用哪个CUDA的步骤。我在系统上有多个CUDA版本,它们通常由模块加载命令调用

任何想法都欢迎

干杯


Tags: 模块用户文档功能版本标志tensorflow系统