坑太多了,踩了很久,给出最佳方案:
- 环境汇总:
- CUDA9.0
- Anaconda
- Python3.6 (conda)
- PyTorch1.0 源码编译
- LibTorch 官网下载
- OpenCV 4.0.0 源码编译
-
步骤:
-
确认CUDA9.0是否安装
-
安装anaconda,确认which python3.6在anaconda目录下
-
使用anaconda包管理,源码编译安装PyTorch
-
Clone下完整的pytorch目录
git clone –recursive https://github.com/pytorch/pytorch
-
安装PyTorch
python3.6 setup.py install
-
安装完毕退出该文件夹:
cd ../
然后运行这个不报错:
python
import torch torch.__version__
-
- 源码编译安装TorchVision
-
Clone目录
-