问题描述
人脸重构代码跑到十几个人或者三十几个人就报错了
解决方法
在脚本中设置环境变量:
import os
os.environ["TORCH_USE_CUDA_DSA"] = "1"
或者:
终端:TORCH_USE_CUDA_DSA=1 python your_script.py
问题描述
人脸重构代码跑到十几个人或者三十几个人就报错了
解决方法
在脚本中设置环境变量:
import os
os.environ["TORCH_USE_CUDA_DSA"] = "1"
或者:
终端:TORCH_USE_CUDA_DSA=1 python your_script.py