在启用GPU的计算机上安装基于CPU的库版本

Question

我想在配备nVIDIA Tesla GPU的服务器上安装CPU版PyTorch。它是否可行或者我是否只能在此服务器上安装GPU版本（使用CUDA）以使PyTorch正常运行？

Answer 1

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