OpenNMT-py使用GPU模式进行训练和翻译

最新推荐文章于 2025-06-06 09:05:37 发布

原创

最新推荐文章于 2025-06-06 09:05:37 发布 · 8.2k 阅读

9 ·

CC 4.0 BY-SA版权

文章标签：

#OpenNMT #GPU #Torch #torchtext #machine translation

本文档介绍了在拥有双NVIDIA GeForce GTX 1080 Ti GPU的Windows 10系统上，如何配置环境，解决TorchText 0.2.3与Torch 0.4.0不兼容问题，以及OpenNMT-py训练过程中遇到的dtype参数错误，并提供了GPU训练的基本步骤。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

硬件情况

CPU：Intel(R) Core(TM) i7-7700K CPU @ 4.2GHz

GPU: NVIDIA GeForce GTX 1080 Ti * 2

RAM: 32G

OS: Windows 10 Pro

环境准备

Python

Python 选择安装3.6版本，安装好以后配置环境变量。

Anaconda3

Anaconda3选择Anaconda3-5.2.0版本。在使用PyTorch时建议在Anaconda3里建一个新的环境就命名叫“PyTorch”，我们将需要用的一些包都安装到这个环境中去，这样在使用中不会干涉其他环境。在Anaconda3中打开指定环境的终端可以直接操作该环境。

Torch

Torch在Anaconda3里进行安装就可以。选择的版本是0.4.0版本。打开Anaconda3，选择新建的环境“PyTorch”，右击选择打开终端，在该终端中使用conda命令来进行安装。

TorchText

TorchText是在安装完Torch之后进行安装的，使用Anaconda3进行安装。但是需要注意的是与Torch 0.4.0所匹配的 TorchText 0.2.3 版本存在问题，会导致翻译报错。错误是“TypeError: make_src() takes 2 positional arguments but 3 were given”。遇到这个错误目前的方法如下。