ubuntu16.04安装pytorch踩坑

本文只针对我在安装过程中遇到的问题,仅供参考。具体安装教程请参考pytorch官网(https://pytorch.org/get-started/locally/)。 第一次安装:因为电脑已经安装了CUDA,所以直接按照官网的指令安装,如下图蓝色框所示所示: 网速一定要好。显示安装成功之后,做测试。 为确保正确安装PyTorch,在这里,我们将构造一个随机初始化张量。如下图所示,可以正确运行。
相关文章
相关标签/搜索