reml-lab / mTAN

Code for "Multi-Time Attention Networks for Irregularly Sampled Time Series", ICLR 2021.
MIT License
113 stars 22 forks source link

Not executable without NVIDIA GPU #6

Closed mzabaletasar closed 2 years ago

mzabaletasar commented 2 years ago

When running the tan_interpolation.py instruction, I get the following:

50078 (1000, 20) (1000, 20) (1000, 100) (1000, 20, 3) [[ 0.8516054 1. 0.09 ] [ 0.95830714 1. 0.2 ] [ 1.33468433 1. 0.25 ] [ 1.95121209 1. 0.37 ] [ 1.88823672 1. 0.39 ] [ 1.18921462 1. 0.46 ] [ 1.03273212 1. 0.47 ] [ 0.70050108 1. 0.49 ] [ 0.26575831 1. 0.64 ] [ 0.42114019 1. 0.69 ] [ 0.33637057 1. 0.72 ] [ 0.08979964 1. 0.77 ] [ 0.01292361 1. 0.79 ] [-0.01584657 1. 0.8 ] [-0.03790797 1. 0.81 ] [-0.05342294 1. 0.82 ] [-0.04604355 1. 0.87 ] [-0.03038688 1. 0.88 ] [-0.03038688 1. 0.88 ] [ 0.27021355 1. 0.99 ]] (800, 20, 3) (200, 20, 3) parameters: 49400 64381 Traceback (most recent call last): File "tan_interpolation.py", line 129, in out = rec(torch.cat((subsampled_data, subsampled_mask), 2), subsampled_tp)

(some more callback traces)

and then,

AssertionError: Found no NVIDIA driver on your system. Please check that you have an NVIDIA GPU and installed a driver from http://www.nvidia.com/Download/index.aspx

Is there a way to run these scripts without having a NVIDIA GPU?