mirror of
https://github.com/saymrwulf/pytorch.git
synced 2026-05-14 20:57:59 +00:00
10 lines
339 B
Bash
Executable file
10 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!!"
|