参考自源码编译pytorch
由于要紧急测试pytorch的代码,实验室环境下无anaconda,打算直接在无anaconda的环境下编译pytorch的源码
1、首先安装yaml依赖
sudo apt install python-yaml python3-yaml
2、git clone pytorch 的源码
git clone https://github.com/pytorch/pytorch.git
git clone https://github.com/pytorch/vision.git
3、在pytorch 文件夹下运行setup.py
sudo python3 setup.py install
会提示需要进行 git submodule update --init (主要作用是子模块的路径更新,这步必须要做)
git submodule update --init
sudo python3 setup.py install
安装pytorch 成功
4、安装torchvision
在vision文件夹下运行setup.py
sudo python3 setup.py install
安装成功
5、注意事项:安装成功后检测是否可以导入时,需要新开一个终端进行import