安装过程
之前尝试过各种安装方式,最后都失败了,没办法只能采用源码进行安装,系统是ubuntu 16.04
1 下载源码安装torch
git clone https://github.com/pytorch/pytorch.git
git clone https://github.com/pytorch/vision.git
cd pytorch/
#如果是安装CPU版本的话在加一个声明
export NO_CUDA=1
git submodule update --init ---路径更新
sudo python3 setup.py install
这里需要注意一下,这个 python3 p y t h o n 3 ,我是安装了conda的python同时,加了环境变量,但是最后发现 torch t o r c h 装在了系统自带的 python3 p y t h o n 3 下面,只能重新安装一下, python p y t h o n 命令使用了绝对路径,如下
sudo /home/xxxx/anaconda3/binpython3 setup.py install
2 安装vision
cd vision
sudo /home/xxxx/anaconda3/binpython3 setup.py install
安装完了,需要重新开一个终端, import i m p o r t 就可以了。