本篇博客详细介绍了如何在Ubuntu16.04上安装PyTorch，针对CUDA 8.0环境，选择了Python3.5对应的whl文件，并通过pip3进行安装。同时，还指导了如何安装torchvision，以及解决可能出现的numpy版本问题。安装完成后，通过代码验证了GPU的可用性。
进入PyTorch官网：
PyTorch
在PyTorch主页上右下角点击“Click here for previous versions of PyTorch”，如下图：
然后在弹出的界面，根据你自身装的CUDA版本去选择对应的whl文件进行下载：
要用GPU的话，就要选择含cu的版本，我的电脑安装的是CUDA 8，然后我想安装在系统的Python3.5上，所以选择如下：
其中cp35代表python3.5，同理，cp27就是python2.7
前面的0.4.0或者0.3.1就是PyTorch的版本数，建议使用0.3.1，因为现在很多开源的代码都是基于0.3的，不同版本的函数有些不一样
进入PyTorch的下载目录，使用pip3命令安装：（python3是用pip3，python2就直接用pip，根据自己的情况，注意这点不同）
sudo pip3 install torch-0.3
最低0.47元/天 解锁文章
399
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
