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
相关文章
相关标签/搜索