Can I run my jupyter notebook that contains seaborn code on GPU?

后端 未结 3 2099
一整个雨季
一整个雨季 2021-01-25 04:08

I have CUDA compatible GPU (Nvidia GeForce 1060) in my system. While analyzing a bigger dataset, I often have to use pair plot function of the seaborn library, it consumes a lot

3条回答
  •  忘了有多久
    2021-01-25 04:36

    Not sure whether your GPU is supported but there is now (Q3 2020) options to do data manipulation on GPU using libraries such as cudf or cupy.

    I am just starting down this path and from little I've seen, you will have to do some "extra" work in transferring results to a format that Seaborn can handle, but calculations with cudf (and I assume cupy) are much faster (I've seen 5x to 25x improvements so far, and read of even more extreme cases).

    You will have to write a little more code though...

提交回复
热议问题