cc_staff
282
edits
PyTorch: Difference between revisions
Jump to navigation
Jump to search
m
no edit summary
mNo edit summary |
mNo edit summary |
||
| Line 1,612: | Line 1,612: | ||
#!/bin/bash | #!/bin/bash | ||
#SBATCH --nodes 1 | #SBATCH --nodes 1 | ||
#SBATCH --gres=gpu: | #SBATCH --gres=gpu:2 # Request 2 GPU "generic resources”. | ||
#SBATCH --tasks-per-node=2 # Request 1 process per GPU. You will get 1 CPU per process by default. Request more CPUs with the "cpus-per-task" parameter to enable multiple data-loader workers to load data in parallel. | #SBATCH --tasks-per-node=2 # Request 1 process per GPU. You will get 1 CPU per process by default. Request more CPUs with the "cpus-per-task" parameter to enable multiple data-loader workers to load data in parallel. | ||
#SBATCH --mem=32G | #SBATCH --mem=32G | ||