Deep3DFaceRecon_pytorch
Deep3DFaceRecon_pytorch copied to clipboard
ValueError: Unknown CUDA arch (8.9) or GPU not supported
File "E:\Deep3DFaceRecon_pytorch\models\base_model.py", line 162, in test self.forward() File "E:\Deep3DFaceRecon_pytorch\models\facerecon_model.py", line 140, in forward self.pred_vertex, self.facemodel.face_buf, feat=self.pred_color) File "E:\anaconda3\envs\deep3d_pytorch\lib\site-packages\torch\nn\modules\module.py", line 722, in _call_impl result = self.forward(*input, **kwargs) File "E:\Deep3DFaceRecon_pytorch\util\nvdiffrast.py", line 61, in forward self.ctx = dr.RasterizeGLContext(device=device) File "E:\anaconda3\envs\deep3d_pytorch\lib\site-packages\nvdiffrast\torch\ops.py", line 221, in init self.cpp_wrapper = _get_plugin(gl=True).RasterizeGLStateWrapper(output_db, mode == 'automatic', cuda_device_idx) File "E:\anaconda3\envs\deep3d_pytorch\lib\site-packages\nvdiffrast\torch\ops.py", line 118, in _get_plugin torch.utils.cpp_extension.load(name=plugin_name, sources=source_paths, extra_cflags=opts, extra_cuda_cflags=opts+['-lineinfo'], extra_ldflags=ldflags, with_cuda=True, verbose=False) File "E:\anaconda3\envs\deep3d_pytorch\lib\site-packages\torch\utils\cpp_extension.py", line 974, in load keep_intermediates=keep_intermediates) File "E:\anaconda3\envs\deep3d_pytorch\lib\site-packages\torch\utils\cpp_extension.py", line 1179, in _jit_compile with_cuda=with_cuda) File "E:\anaconda3\envs\deep3d_pytorch\lib\site-packages\torch\utils\cpp_extension.py", line 1272, in _write_ninja_file_and_build_library with_cuda=with_cuda) File "E:\anaconda3\envs\deep3d_pytorch\lib\site-packages\torch\utils\cpp_extension.py", line 1593, in _write_ninja_file_to_build_library cuda_flags = common_cflags + COMMON_NVCC_FLAGS + _get_cuda_arch_flags() File "E:\anaconda3\envs\deep3d_pytorch\lib\site-packages\torch\utils\cpp_extension.py", line 1414, in _get_cuda_arch_flags raise ValueError("Unknown CUDA arch ({}) or GPU not supported".format(arch)) ValueError: Unknown CUDA arch (8.9) or GPU not supported
my win11 cuda 8.7, why it is 8.9?