Commit 8885584b authored by chenpangpang's avatar chenpangpang
Browse files

feat: 初始提交

parent 7e6e7f2b
#!/bin/bash
docker run --rm --platform=linux/amd64 --gpus all $1 python -c \
'import torch; \
print("torch version: ", torch.__version__); \
print("torch cuda available: ", torch.cuda.is_available()); \
import torchvision; \
print("torchvision version: ", torchvision.__version__); \
import torchaudio; \
print("torchaudio version: ", torchaudio.__version__);
'
\ No newline at end of file
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment