diff --git a/RaspberryPi_JetsonNano/python/setup.py b/RaspberryPi_JetsonNano/python/setup.py
index 8305abf..3f619d3 100644
--- a/RaspberryPi_JetsonNano/python/setup.py
+++ b/RaspberryPi_JetsonNano/python/setup.py
@@ -1,10 +1,19 @@
-import sys
+import sys, os
 from setuptools import setup
+
+dependencies = ['Pillow', 'numpy']
+
+if os.path.exists('/sys/bus/platform/drivers/gpiomem-bcm2835'):
+    dependencies += ['RPi.GPIO', 'spidev']
+else:
+    dependencies += ['Jetson.GPIO']
+
 setup(
     name='waveshare-epd',
     description='Waveshare e-Paper Display',
     author='Waveshare',
     package_dir={'': 'lib'},
     packages=['waveshare_epd'],
+    install_requires=dependencies,
 )