请问一下函数计算使用GPU实例,CUDA版本是多少,安装Pytorch需要安装哪个版本?
参考 https://github.com/devsapp/start-fc-gpu/tree/main/fc-http-gpu-inference-torchhub-cv-mobilenet-v2
此答案来自钉钉群“阿里函数计算官网客户"
目前阿里云函数计算支持使用GPU实例运行函数计算,CUDA版本是最新的11.2版本。安装Pytorch时需要使用与CUDA版本对应的Pytorch版本,建议使用Pytorch官方提供的CUDA版本和Pytorch版本对照表,选择相应的版本进行安装。同时,也需要根据GPU实例规格进行选择和优化,例如可以提前安装好PyTorch和相关依赖,避免在函数运行时安装过程带来的等待时间。
在阿里云函数计算中,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
版权声明:本文内容由阿里云实名注册用户自发贡献,版权归原作者所有,阿里云开发者社区不拥有其著作权,亦不承担相应法律责任。具体规则请查看《阿里云开发者社区用户服务协议》和《阿里云开发者社区知识产权保护指引》。如果您发现本社区中有涉嫌抄袭的内容,填写侵权投诉表单进行举报,一经查实,本社区将立刻删除涉嫌侵权内容。