用python自带IDLE+cuda能不能实现GPU加速?

python学习小白一枚,最近在搞GPU加速,我想问问能不能直接用python自带的IDLE实现GPU加速呢,已经安装好cuda,不想使用tensorflow和其他的框架。

  • 尤及
  • 尤及     2020-05-29 10:49:401楼

    def set_gpus(gpu_index):
        if type(gpu_index) == list:
            gpu_index = ','.join(str(_) for _ in gpu_index)
        if type(gpu_index) ==int:
            gpu_index = str(gpu_index)
        os.environ["CUDA_VISIBLE_DEVICES"] = gpu_index

    使用gpu都要安装相应的gpu版本,如TensorFlow,pytorch等,检测到gpu,会自动调用

    +添加回复

  • 回复

    同时注意gpu数量就可以啦

    流芳 · 2020-05-29 22:23:40