I recently ran into a weird error, after compiling Pytorch 1.1 with CUDA 10.0 from source:
~/anaconda3/lib/python3.7/site-packages/pynvml.py in _LoadNvmlLibrary()
644 nvmlLib = CDLL("libnvidia-ml.so.1")
645 except OSError as ose:
--> 646 _nvmlCheckReturn(NVML_ERROR_LIBRARY_NOT_FOUND)
647 if (nvmlLib == None):
648 _nvmlCheckReturn(NVML_ERROR_LIBRARY_NOT_FOUND)
~/anaconda3/lib/python3.7/site-packages/pynvml.py in _nvmlCheckReturn(ret)
308 def _nvmlCheckReturn(ret):
309 if (ret != NVML_SUCCESS):
--> 310 raise NVMLError(ret)
311 return ret
312
NVMLError_LibraryNotFound: NVML Shared Library Not Found
Iām running Mac OS High Sierra with the most recent NVIDIA Web Drivers, CUDA 10.0 and CUDNN 7.4.2
Does anyone know, what package I am missing to obtain libnvidia-ml.so.1?
(base) ā ~ clang -v
Apple LLVM version 9.0.0 (clang-900.0.39.2)
Target: x86_64-apple-darwin17.0.0
Thread model: posix
InstalledDir: /Library/Developer/CommandLineTools/usr/bin
Found CUDA installation: /usr/local/cuda, version unknown
(base) ā ~ nvcc --version
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2018 NVIDIA Corporation
Built on Sat_Aug_25_21:08:56_CDT_2018
Cuda compilation tools, release 10.0, V10.0.130
(base) ā ~ python --version
Python 3.7.1
(base) ā ~ pip freeze | grep -e fastai -e torch
fastai==1.0.45
torch==1.1.0a0+fbd690c
torchvision==0.2.1
(base) ā ~ conda --version
conda 4.6.3
Thanks in advance!