gcp上未定义导入火炬

Question

试图在GCP上运行“导入火炬”

我收到消息ModuleNotFoundError: No module named 'torch'

尝试创建具有以下属性的新群集（通过api，而不是终端）：

dataproc->；康达包装->；pytorch==1.7.1，visions==0.7.1

dataproc->；点子包->；标记化器==0.10.1，数据集==1.5.0

群集创建超时，未创建。 即使只是尝试一下，也会发生同样的情况

dataproc->；康达包装->；pytorch==1.7.1

尝试使用以下命令手动在主机上进行安装：

conda install -c pytorch pytorch

差不多一个多小时后，我得到：

Collecting package metadata (current_repodata.json): done
Solving environment: failed with initial frozen solve. Retrying with flexible solve.
Solving environment: failed with repodata from current_repodata.json, will retry with next repodata source.
Collecting package metadata (repodata.json): done
Solving environment -/

谢谢你的帮助和建议 i、 除火炬外，我还需要： 从transformers进口LongformerModel、LongformerTokenizer 从模型导入RedditModel

Answer 1

您只需要使用更新/最新的PyTorch version available in conda-forge channel（现在是1.8.0）：

gcloud dataproc clusters create ${CLUSTER_NAME} \
   image-version=2.0 \
   region=${REGION} \
   properties='^#^dataproc:conda.packages=pytorch==1.8.0,torchvision==0.9.0'

Answer 2

不要使用conda安装，而是尝试使用conda forge。这应该行得通。 https://anaconda.org/conda-forge/pytorch