"torchvision/git@developer.sourcefind.cn:OpenDAS/vision.git" did not exist on "ccb7f45a3570b2175d8e8def66629528d557da3c"
Set `torch.set_deterministic` to false for GPU tests (#3422) (#3435)
Summary: Pull Request resolved: https://github.com/pytorch/vision/pull/3422 In CUDA-10.1 or below it is a no-op, but few of the tests will start to fail if update to 10.2 or newer, for example: ``` torchvision/transforms/transforms.py", line 1043, in forward transformed_tensor = torch.mm(flat_tensor, self.transformation_matrix) RuntimeError: Deterministic behavior was enabled with either `torch.use_deterministic_algorithms(True)` or `at::Context::setDeterministicAlgorithms(true)`, but this operation is not deterministic because it uses CuBLAS and you have CUDA >= 10.2. To enable deterministic behavior in this case, you must set an environment variable before running your PyTorch application: CUBLAS_WORKSPACE_CONFIG=:4096:8 or CUBLAS_WORKSPACE_CONFIG=:16:8. For more information, go to https://docs.nvidia.com/cuda/cublas/index.html#cublasApi_reproducibility ``` Reviewed By: meyering Differential Revision: D26563045 fbshipit-source-id: 65e5fabc8cdb471a72f931df55389ed7c0063e54 Co-authored-by:Nikita Shulga <nshulga@fb.com>
Showing
Please register or sign in to comment