当前位置:网站首页>CUDA versions are inconsistent, and errors are reported when compiling apex

CUDA versions are inconsistent, and errors are reported when compiling apex

2022-07-07 20:05:00 Scrambled eggs for tomatoes

terms of settlement

Comment out setup.py In the code 108 That's ok ,cuda Version verification , Just compile in this way

if "--cuda_ext" in sys.argv:
    from torch.utils.cpp_extension import CUDAExtension
    sys.argv.remove("--cuda_ext")

    if torch.utils.cpp_extension.CUDA_HOME is None:
        raise RuntimeError("--cuda_ext was requested, but nvcc was not found.  Are you sure your environment has nvcc available?  If you're installing within a container from https://hub.docker.com/r/pytorch/pytorch, only images whose names contain 'devel' will provide nvcc.")
    else:
        #  Comment out this sentence 
        # check_cuda_torch_binary_vs_bare_metal(torch.utils.cpp_extension.CUDA_HOME)

原网站

版权声明
本文为[Scrambled eggs for tomatoes]所创,转载请带上原文链接,感谢
https://yzsam.com/2022/188/202207071755028628.html