search

jiazhihao / TASO

The Tensor Algebra SuperOptimizer for Deep Learning
Apache License 2.0
692 stars 90 forks source link

Compiling TASO #56

Open shersoni610 opened 4 years ago

shersoni610 commented 4 years ago

Hello,

From the Dockerfile, I see that you are using Cuda 10.0 and Ubuntu 16. Can I use the latest versions instead? WIll it work?

mikeseven commented 4 years ago

yes I use cuda 10.2 on ubuntu 18.04.