Skip to content

GitLab

Unverified Commit eaf25af6 authored by Jintao Lin's avatar Jintao Lin Committed by GitHub
Browse files

use CUDA_HOME from parrots_wrapper (#678)

parent d2ca6ceb
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
mmcv/utils/env.py
View file @ eaf25af6
...@@ -48,7 +48,7 @@ def collect_env(): ...@@ -48,7 +48,7 @@ def collect_env():
for name, device_ids in devices.items(): for name, device_ids in devices.items():
env_info['GPU ' + ','.join(device_ids)] = name env_info['GPU ' + ','.join(device_ids)] = name
from torch.utils.cpp_extension import CUDA_HOME from mmcv.utils.parrots_wrapper import CUDA_HOME
env_info['CUDA_HOME'] = CUDA_HOME env_info['CUDA_HOME'] = CUDA_HOME
if CUDA_HOME is not None and osp.isdir(CUDA_HOME): if CUDA_HOME is not None and osp.isdir(CUDA_HOME):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment