如何指定Python版本3.6.8用于PyTorch Estimator(conda_packages不足)

2024-04-24 10:01:19 发布

您现在位置:Python中文网/ 问答频道 /正文

我需要在Azure机器学习下运行python脚本,使用python=3.6.8(不是默认的3.6.2)。我正在使用AML“PyTorch()”估计器,将“conda\u packages”arg设置为[“python==3.6.8”]。你知道吗

我依赖这个文档页面来获取PyTorch估算器:

https://docs.microsoft.com/en-us/python/api/azureml-train-core/azureml.train.dnn.pytorch?view=azure-ml-py

当我的脚本运行时,我打印出“系统版本”并查看它仍然设置为Python3.6.2

python: 3.6.2 | packaged by conda-forge | (default, Jul 23 2017, 22:59:30) 
[GCC 4.8.2 20140120 (Red Hat 4.8.2-15)]

我希望看到python3.6.8,因为我在PyTorch估计器的conda_包arg中指定了它。你知道吗

我还尝试将“python==3.6.8”从conda\u包移到pip\u包,但收到一个错误,说明pip找不到该包。你知道吗

仅供参考,我在pip\u软件包中指定了另一个软件包,并且在这个过程中确实正确安装了它。似乎“conda_packages”参数的值没有被使用(我的作业的AML日志中没有提到conda或python安装错误)。你知道吗


Tags: pip文档https脚本机器packages错误arg
2条回答

另一个选项是使用正确的python版本指定conda依赖文件conda\u dependencies\u file\u路径。下面的文档概述了如何做到这一点的详细文档。一旦您指定conda\u dependencies\u file\u path,它就会覆盖pip\u包和conda\u包,因此我建议将所有包放在conda dependency文件中

https://docs.microsoft.com/en-us/python/api/azureml-core/azureml.core.conda_dependencies.condadependencies?view=azure-ml-py

提交培训作业时,azureml在Docker容器中的conda环境中运行脚本。PyTorch容器安装了以下依赖项。你知道吗

依赖项
PyTorch 1.0/1.1/1.2型 Python 3.6.2 CUDA(仅限GPU映像)10.0 cuDNN(仅限GPU映像)7.6.3 NCCL(仅限GPU映像)2.4.8 azureml默认最新OpenMpi 3.1.2 horovod 0.16.1 miniconda 4.5.11 torch 1.0/1.1/1.2 torchvision 0.2.1 git 2.7.4 tensorboard 1.14 future 0.17.1 Docker映像扩展了Ubuntu 16.04。你知道吗

要安装附加的依赖项,可以使用pip\u packages或conda\u packages参数。或者,您可以指定pip\u requirements\u file或conda\u dependencies\u file参数。或者,您可以构建自己的映像,并将自定义的\u docker \u image参数传递给estimator构造函数。你知道吗

我建议您更新conda\u dependencies\u file参数,看看它是否有效。你知道吗

引用:https://docs.microsoft.com/en-us/azure/machine-learning/service/how-to-train-pytorch#create-a-pytorch-estimator

相关问题 更多 >