We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 317f18c commit 6b7df0bCopy full SHA for 6b7df0b
internlm/utils/common.py
@@ -250,9 +250,8 @@ def enable_pytorch_expandable_segments():
250
251
252
def check_cuda_env():
253
- max_connections = os.getenv("CUDA_DEVICE_MAX_CONNECTIONS")
254
- assert max_connections is not None, "Env var CUDA_DEVICE_MAX_CONNECTIONS has not been set, please set it to 1!"
255
- assert max_connections == '1', "Env var CUDA_DEVICE_MAX_CONNECTIONS is set to {}, but it should be set to 1!".format(max_connections)
+ if os.getenv("CUDA_DEVICE_MAX_CONNECTIONS") is None:
+ logger.warning("Env var CUDA_DEVICE_MAX_CONNECTIONS has not be set, please note this!")
256
257
258
class DummyProfile:
0 commit comments