Inconsistent Packages version in Prerequisite vs yml

[npatel-ah]

Hello,

I have a query regarding CUDA and CUDNN versions. The prerequisite is "CUDA 8.0 with cudnn 5.1", however, the yml file says otherwise

  - cuda90=1.0=h6433d27_0
  - pytorch=0.3.1=py36_cuda9.0.176_cudnn7.0.5_2

I ignored the Prerequisites and created an environment with the yml file which ran fine also I was able to run the preprocess.py step but when I ran "main.py" I got error :
AssertionError: Found no NVIDIA driver on your system.
Any help on this is appreciated.

Best