iicd/autodl-projects
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update tests for torch/cuda

Browse Source
This commit is contained in:
D-X-Y
2021-03-30 12:05:52 +00:00
parent c2270fd153
commit e5ec43e04a
12 changed files with 386 additions and 248 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tests/test_torch.sh Normal file
View File

@@ -0,0 +1,4 @@
# bash ./tests/test_torch.sh
pytest ./tests/test_torch_gpu_bugs.py::test_create -s
CUDA_VISIBLE_DEVICES="" pytest ./tests/test_torch_gpu_bugs.py::test_load -s