开发者社区 > 云原生 > Serverless > 正文

请问一下函数计算使用GPU实例,CUDA版本是多少,安装Pytorch需要安装哪个版本?

请问一下函数计算使用GPU实例,CUDA版本是多少,安装Pytorch需要安装哪个版本?

展开
收起
山海行 2023-03-14 16:36:37 531 0
3 条回答
写回答
取消 提交回答
  • 参考 https://github.com/devsapp/start-fc-gpu/tree/main/fc-http-gpu-inference-torchhub-cv-mobilenet-v2

    回答5.png

    此答案来自钉钉群“阿里函数计算官网客户"

    2023-03-14 17:27:21
    赞同 展开评论 打赏
  • 月移花影,暗香浮动

    目前阿里云函数计算支持使用GPU实例运行函数计算,CUDA版本是最新的11.2版本。安装Pytorch时需要使用与CUDA版本对应的Pytorch版本,建议使用Pytorch官方提供的CUDA版本和Pytorch版本对照表,选择相应的版本进行安装。同时,也需要根据GPU实例规格进行选择和优化,例如可以提前安装好PyTorch和相关依赖,避免在函数运行时安装过程带来的等待时间。

    2023-03-14 17:11:21
    赞同 展开评论 打赏
  • 发表文章、提出问题、分享经验、结交志同道合的朋友

    在阿里云函数计算中,GPU 实例默认安装了 NVIDIA CUDA 工具包和 NVIDIA 驱动程序。当前支持的 CUDA 版本为 CUDA 10.2。

    想在函数计算中使用 PyTorch,需要安装适用于 CUDA 10.2 的 PyTorch 版本。

    pip install torch torchvision torchaudio -f https://download.pytorch.org/whl/cu102/torch_stable.html
    
    
    2023-03-14 16:57:31
    赞同 展开评论 打赏

快速交付实现商业价值。

相关产品

  • 函数计算
  • 相关电子书

    更多
    Hologres Serverless之路:揭秘弹性计算组 立即下载
    Serverless开发平台: 让研发效能再提升 立即下载
    Serverless 引领云上研发新范式 立即下载