orasanen / ALICE

Automatic LInguistic Unit Count Estimator (ALICE)
44 stars 11 forks source link

GPU requires CUDA 9.X #13

Open cvanpay opened 4 years ago

cvanpay commented 4 years ago

Because it uses Tensorflow 1.10.1, this requires CUDA 9.0. In the ALICE.yml it just specifies cudatoolkit, which is fine for MacOS because that is still the main version, but not for Linux. https://anaconda.org/anaconda/cudatoolkit

Should be safe to put cudatoolkit=9.0 in the ALICE.yml.

Edited: Only CUDA 9.0 will work with Tensorflow 1.10.1.