在支持 GPU 的机器上安装基于 CPU 的库版本

Question

我想在配备 nVIDIA Tesla GPU 的服务器上安装 CPU 版本的 PyTorch。它会工作还是我只能在此服务器上安装 GPU 版本（带有 CUDA）以使 PyTorch 正常运行？

Answer 1

具有 GPU 支持的 PyTorch 版本也适用于 CPU（但 cpu 神经网络训练非常慢）。所以你可以安装GPU版本。确保安装使用正确的 cuda 版本（cuda 7.5、cuda 8.0 或 cuda 9.0）编译的 PyTorch。