1、装好cuda9.0,将cudnn7的文件拷到对应位置
2、安装anaconda,安装anaconda时选择添加环境变量复选框
3、安装pytorch,使用下载好的镜像
进入anaconda,conda create --name pytorch-gpu python=3.6 (TensorFlow使用conda create --name tensorflow-gpu python=3.6)
4、activate pytorch-gpu(TensorFlow使用activate tensorflow-gpu)
或者进入anaconda直接pip install pytorch也可以,如下
pip install D:\torch....Tab键(选择预先下载好的torch本地包)
注意:pytorch安装也可以直接在anaconda的prompt中直接 pip install D:\tensoflow....Tab键
5、安装opencv: pip install opencv-python
6、问题:出现ModuleNotFoundError: No module named 'numpy.core._multiarray_umath'
pip install -U numpy
7、问题:ModuleNotFoundError: No module named 'torchvision'
pip install torchvision
8、问题:RuntimeError: cuDNN error: CUDNN_STATUS_EXECUTION_FAILED
安装pytorch1.1.0+cuda10.0+cudnn7.4.1
代码中加入cudnn.benchmark = True,不然会CUDNN_STATUS_EXECUTION_FAILED
9、问题:TypeError: Expected Ptr<cv::UMat> for argument '%s'
把tensorflow和pytorch装在一起引起的冲突,重新安装
10、问题:ImportError: cannot import name 'Optional
版本不匹配的问题。
pip install --no-deps torchvision==0.4.1