详见: https://github.com/pytorch/pytorch/issues/22844
安装和代码中的CUDA_HOME调用函数逻辑不一致,在多CUDA环境中出现bug。
保险的做法是在设置PATH, LD_LIBRARY_PATH等环境变量时顺带把CUDA_HOME也设置了。

windows应该是CUDA_PATH环境变量。

Logo

旨在为数千万中国开发者提供一个无缝且高效的云端环境,以支持学习、使用和贡献开源项目。

更多推荐