Skip to content

GitLab

Commit 1e0f9743 authored by chenpangpang's avatar chenpangpang
Browse files

feat: 输出os信息

parent 9fea35de
Hide whitespace changes
Inline Side-by-side
Showing with 12 additions and 10 deletions
script/1_base_test.sh
View file @ 1e0f9743
#!/bin/bash #!/bin/bash
docker run --rm --platform=linux/amd64 --gpus all $1 cat /etc/issue && python -c \ docker run --rm --platform=linux/amd64 --gpus all $1 python -c \
'import sys; \ "import os; \
print("python version: ", sys.version); \ os.system(\"cat /etc/issue\"); \
import sys; \
print(\"python version: \", sys.version); \
import torch; \ import torch; \
print("torch version: ", torch.__version__); \ print(\"torch version: \", torch.__version__); \
print("torch cuda available: ", torch.cuda.is_available()); \ print(\"torch cuda available: \", torch.cuda.is_available()); \
print("torch cuda version: ", torch.version.cuda); \ print(\"torch cuda version: \", torch.version.cuda); \
print("torch cudnn version: ",torch.backends.cudnn.version()); \ print(\"torch cudnn version: \",torch.backends.cudnn.version()); \
import torchvision; \ import torchvision; \
print("torchvision version: ", torchvision.__version__); \ print(\"torchvision version: \", torchvision.__version__); \
import torchaudio; \ import torchaudio; \
print("torchaudio version: ", torchaudio.__version__); print(\"torchaudio version: \", torchaudio.__version__);
' "
\ No newline at end of file \ 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