gpt4 book ai didi

numba - 如何在协作室中使用numba

转载 作者:行者123 更新时间:2023-12-04 13:56:47 29 4
gpt4 key购买 nike

有人尝试在Google合作伙伴中使用numba吗?我只是不知道如何在此环境中进行设置。
此刻,我陷入了错误library nvvm not found

最佳答案

将此代码复制到单元格中。这个对我有用。

!apt-get install nvidia-cuda-toolkit
!pip3 install numba

import os
os.environ['NUMBAPRO_LIBDEVICE'] = "/usr/lib/nvidia-cuda-toolkit/libdevice"
os.environ['NUMBAPRO_NVVM'] = "/usr/lib/x86_64-linux-gnu/libnvvm.so"

from numba import cuda
import numpy as np
import time

@cuda.jit
def hello(data):
data[cuda.blockIdx.x, cuda.threadIdx.x] = cuda.blockIdx.x

numBlocks = 5
threadsPerBlock = 10

data = np.ones((numBlocks, threadsPerBlock), dtype=np.uint8)

hello[numBlocks, threadsPerBlock](data)

print(data)

关于numba - 如何在协作室中使用numba,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/48811248/

29 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com