未验证 提交 194b0043 编写于 作者: M Maxime 提交者: GitHub

fix: linker issues in conda environments #3929 (#4235)

* fix: linker issues in conda environments #3929

* ignore: re-ordering

* Update builder.py
上级 57d629a1
......@@ -702,6 +702,8 @@ class TorchCPUOpBuilder(CUDAOpBuilder):
if not self.build_for_cpu:
if not self.is_rocm_pytorch():
CUDA_LIB64 = os.path.join(torch.utils.cpp_extension.CUDA_HOME, "lib64")
if not os.path.exists(CUDA_LIB64):
CUDA_LIB64 = os.path.join(torch.utils.cpp_extension.CUDA_HOME, "lib")
else:
CUDA_LIB64 = os.path.join(torch.utils.cpp_extension.ROCM_HOME, "lib")
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册