pytorch环境的配置

pytorch环境配置比tensorflow简单太多了。 装好合适的nvidia显卡驱动和cuda后，直接去官网下载 pytorch官网

选择对应的版本，我这里使用的pip下载，直接赋值命令到命令行即可

在程序中：

torch.cuda.is_available()

若返回true即为成功。