-
Notifications
You must be signed in to change notification settings - Fork 105
Description
Hello, thanks for providing the code for your paper!
I have been unable to run either the benchmark_agent.py code (with the pre-trained model-10.th), or the data_collector.py script. Both scripts seem to be freezing at some point in the import process, which I haven't been able to resolve. They don't throw errors (or print anything besides the pygame message) but seem to hang indefinitely.
I had to make a couple of modifications to the installed dependencies, because it seems that the RTX GPU family is not compatible with CUDA 9.0 or below (or with any packages that are built with CUDA 9.0 or below), as per pytorch/pytorch#17543, and because I ran into CUDA warnings and CUDNN errors when using the original dependencies (that were built with CUDA 8.0)
The relevant versions of dependencies on my system are as follows:
| Package | Version |
|---|---|
| cudatoolkit | 10.1.243 |
| pytorch | py3.5_cuda10.1.243_cudnn7.6.3_0 |
| cudnn | 7.6.5 |
If anyone has guidance on how to debug/resolve this issue, I'd really appreciate it. Thanks so much!
My set-up:
GPU: GeForce RTX 2080 with Max-Q
CUDA version: 10.1