pytorch小坑:需设置CUDA_HOME环境变量,保证全局CUDA环境一致
·
详见: https://github.com/pytorch/pytorch/issues/22844
安装和代码中的CUDA_HOME
调用函数逻辑不一致,在多CUDA环境中出现bug。
保险的做法是在设置PATH
, LD_LIBRARY_PATH
等环境变量时顺带把CUDA_HOME
也设置了。
windows应该是CUDA_PATH
环境变量。
更多推荐
已为社区贡献10条内容
所有评论(0)