本文档介绍了在拥有双NVIDIA GeForce GTX 1080 Ti GPU的Windows 10系统上，如何配置环境，解决TorchText 0.2.3与Torch 0.4.0不兼容问题，以及OpenNMT-py训练过程中遇到的dtype参数错误，并提供了GPU训练的基本步骤。
硬件情况
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
最低0.47元/天 解锁文章
扫一扫
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
