mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 12:21:27 +01:00
11 lines
339 B
Bash
Executable File
11 lines
339 B
Bash
Executable File
#!/bin/bash
|
|
|
|
set -e
|
|
|
|
python -c 'from caffe2.python import build; from pprint import pprint; pprint(build.build_options)'
|
|
python -c 'from caffe2.python import core, workspace; print("GPUs found: " + str(workspace.NumCudaDevices()))'
|
|
python -c "import onnx"
|
|
python -c "import torch"
|
|
|
|
echo "Caffe2, PyTorch and ONNX installed successfully!!"
|