pytorch小坑:需設置CUDA_HOME環境變量,保證全局CUDA環境一致

詳見: https://github.com/pytorch/pytorch/issues/22844 安裝和代碼中的CUDA_HOME調用函數邏輯不一致,在多CUDA環境中出現bug。 保險的作法是在設置PATH, LD_LIBRARY_PATH等環境變量時順帶把CUDA_HOME也設置了。git windows應該是CUDA_PATH環境變量。github
相關文章
相關標籤/搜索