pytorch-gpu-benchmark
pytorch-gpu-benchmark copied to clipboard
ryujaehun
Using the famous cnn model in Pytorch, we run benchmarks on various gpu.
~/pytorch-gpu-benchmark# python3 benchmark_models.py Traceback (most recent call last): File "/root/pytorch-gpu-benchmark/benchmark_models.py", line 10, in import pandas as pd ModuleNotFoundError: No module named 'pandas' =============== Does this tool support H100(8 gpus) systems?
I tested the performance of my GPU - NVIDIA RTX 5000 Ada Generation Laptop GPU
Running env: docker in wsl2 20.10.16 nvidia pytorch image: nvcr.io/nvidia/pytorch:22.05-py3 Here is the error code ```shell= root@dd288f178ecc:/workspace/pytorch-gpu-benchmark# ./test.sh start benchmark start : 2022/06/28 08:39:29 Number of GPUs on current device...
I am running Ubuntu 20.04.2 with all updates: ``` $ uname -a Linux bengt-desktop 5.11.0-37-generic #41~20.04.2-Ubuntu SMP Fri Sep 24 09:06:38 UTC 2021 x86_64 x86_64 x86_64 GNU/Linux ``` I am...
I tested this after building on ROCm 6.1.2. The tests run wonderfully on my RX 7900 XTX: eitch@eitchtower:~/src/compile_temp/pytorch-gpu-benchmark$ export CUDA_VISIBLE_DEVICES=0 eitch@eitchtower:~/src/compile_temp/pytorch-gpu-benchmark$ ./test.sh AMD gpu benchmarks starting GPU count: 1 hip_fatbin.cpp:...
Generated results for AMD RX 6800. (https://github.com/ryujaehun/pytorch-gpu-benchmark/pull/28) When I have the csv files in results folder and open plot with command `jupyter-notebook plot.ipynb` and execute the code I get following...
Modify test.sh so that both the AMD and Nvidia GPU's are supported for running benchmarks. Add also the csv results for AMD RX 6800. Tested on pytorch 2.3.0 and pytorch...
AMD's rocm platform has rocm-smi tool that is similar to nvidia-smi tool. Add suport to test.sh script to first check the existence of nvidia-smi and nvidia-smi before using them to...