##PyTorch for GPGPU-Sim using CUDNN Contributor: Jonathan Lew, University of British Columbia, Dept. of Electrical and Computer Engineering
Welcome to PyTorch for GPGPU-Sim!
This repo is based on the popular PyTorch framework, release v0.4.1, the latest release as of Aug 17, 2018. It can be found at https://github.com/pytorch/pytorch
Note: Only Linux (Ubuntu 16.04, 18.04) install has been tested. Instructions for other operating systems are from the pytorch repo and are left here for your convenience.
Install
- NVIDIA CUDA 7.5 or above
- NVIDIA cuDNN v6.x or above
If you want to build on Windows, Visual Studio 2017 and NVTX are also needed.
Follow steps 1 and 2 here.
git clone [this repo]
cd pytorch-gpgpu-sim
git -c submodule."third_party/nervanagpu".update=none submodule update --initThe Nervana repository no longer exists, so git clone --recursive can not be used. To work around this, we manually disable pulling from that particular repository, while updating the rest of the modules.
On Linux
The following variables must be set:
export CUDNN_INCLUDE_DIR=<path of directory where cudnn.h is located>
export CUDNN_LIBRARY=<path of static cudnn library>
These variables let Pytorch know where to look for CuDNN files during installation.
export TORCH_CUDA_ARCH_LIST="6.1+PTX"
This variable sets the sm and compute flags used to build anything CUDA related. Without setting it, a script will be ran to to probe the GPU for it. It is unclear if the GPU-probing library will use GPGPU-Sim or hardware, but it cannot detect GPGPU-Sim configuration file anyways. If it does not detect any GPUs, it defaults to a full list, which includes “6.1+PTX” (the flag we want.) Manually setting this will reduce the number of PTX files required to parse later.
If Torchvision is not installed, install that first. This will also install NumPy if it was not installed already, as well as the latest version of Pytorch. Since the current version of Pytorch that supports gpgpusim is 0.4 (as opposed to 1.x), some functionalities in the newest version of Torchvision are not compatible with the older version of pytorch. As a result, we recommend installing Torchvision 0.2.2.
pip install torchvision==0.2.2
Installing Torchvision would install the latest Pytorch as well, so manually uninstall this before building the one compatible with gpgpu-sim.
pip uninstall torch
If Ubuntu 18.04 is used, either disable the gcc warnings in CUDA, or modify the CMake file cmake/MiscCheck.cmake to disable the check for gcc versions (specifically lines 29-33 here.)
After modifying the CMake file, you may have to update your PATH variable to point to a directory with gcc-5 and g++-5 (if the gcc/g++ in /usr/bin is a newer version of gcc/g++. This would disable warnings from cudnn.h.
Next, you may have to modify tools/build_pytorch_lib.sh to specify the CUDA host compiler. This can be done by adding -DCUDA_HOST_COMPILER='<path to gcc-5>' \ to the set of flags at every line that starts with ${CMAKE VERSION} (more specifically here and here. For the latter, the script warns you to add flags elsewhere, but it should work regardless.) This would disable the warnings from cmake when it uses nvcc.
Next, since we manually specified which gcc version to use, cmake might not pick up on it and may include compiler flags not supported in gcc-5, so line 231 in CMakeLists.txt may need to be commented out (here.)
Lastly, to enable running other algorithms, cuBLAS must be statically linked (for example, FFT directly calls cuBLAS kernels), which can be done by changing line 255 to the path to the static cuBLAS library in cmake/public/cuda.cmake (here.)
python setup.py installUse the following if you encounter root permission error:
python setup.py install --userSet PYTORCH_BIN to point to the shared library generated by PyTorch.
Export PYTORCH_BIN=<pytorch directory>/torch/lib/libcaffe2_gpu.so
If the shared library is not in that directory, it is possible that CUDA paths were not set, and Pytorch was not built with CUDA support. In that case, uninstall the current pytorch version using pip uninstall, and restart from GPGPU-Sim setup.
Try out your application.
A tested application that ran was the MNIST example in pytorch’s repository here. For now, make sure to turn on the deterministic flag and turn off the benchmark flag (add torch.backends.cudnn.deterministic = True and torch.backends.cudnn.benchmark = False after if __name__ == “main”), and make sure to set iteration and epoch to 1, or else it would take potentially months before simulation completes.
Torch's benchmark mode tries and times every convolution algorithm before picking the fastest one to use, so it would take a long time to simulate. In fact, some algorithms do not have PTX implementations, so they cannot be simulated, meaning the simulation itself may abort if benchmark mode is left on. To remedy this, make sure to turn off benchmark mode.
On macOS
MACOSX_DEPLOYMENT_TARGET=10.9 CC=clang CXX=clang++ python setup.py installOn Windows
set "VS150COMNTOOLS=C:\Program Files (x86)\Microsoft Visual Studio\2017\Enterprise\VC\Auxiliary\Build"
set CMAKE_GENERATOR=Visual Studio 15 2017 Win64
set DISTUTILS_USE_SDK=1
REM The following line is needed for Python 2.7, but the support for it is very experimental.
set MSSdk=1
REM As for CUDA 8, VS2015 Update 2 or up is required to build PyTorch. Use the following two lines.
set "PREBUILD_COMMAND=%VS140COMNTOOLS%\..\..\VC\vcvarsall.bat"
set PREBUILD_COMMAND_ARGS=x64
call "%VS150COMNTOOLS%\vcvarsall.bat" x64 -vcvars_ver=14.11
python setup.py installThree pointers to get you started:
