您的位置:首页 > 编程语言 > Python开发

Windows+Pytorch+EfficientDet

2020-06-16 14:31 851 查看

精选30+云产品,助力企业轻松上云!>>>

 

win7+python3.7+cuda10.1+pytorch1.4

(1)安装anaconda【先查看py37对应的anaconda版本】

https://repo.anaconda.com/archive/

https://docs.anaconda.com/anaconda/packages/oldpkglists/

我这里是py37,anaconda就是Anaconda3-5.3.0-Windows-x86_64.exe,默认安装即可。

 

(2)安装ffmpeg【torchvision会用到】

不安装,可能会提示:torchvision installation problem: avcodec-58.dll not found

1):https://ffmpeg.zeranoe.com/builds/ 下载对应版本的动态库

2):将bin文 1000 件添加到系统环境变量

 

(3)安装mkl版本的numpy

不安装,可能会提示:from scipy.linalg import _fblas    

1):到https://www.lfd.uci.edu/~gohlke/pythonlibs/#numpy,

2):根据自己的python版本和numpy版本找到对应的文件下载对应的whl文件

3):pip install 此安装文件   

我这里是:numpy-1.16.4+mkl-cp37-cp37m-win_amd64.whl

上面的网站下载太慢了,可以到这里下载:https://download.csdn.net/download/lilai619/12514864

 

我这里还提示了:

twisted 18.7.0 requires PyHamcrest>=1.9.0, which is not installed.

直接安装就可以了:

pip install -i https://mirrors.aliyun.com/pypi/simple/ PyHamcrest==1.9.0

 

(4)安装pytorch, torchvision

1):离线下载torch,torchvision包

https://download.pytorch.org/whl/torch_stable.html

 

 

2):使用pip install安装   

 

 

pip install torch-1.4.0-cp37-cp37m-win_amd64.whl
pip install torchvision-0.5.0-cp37-cp37m-win_amd64.whl

(5)剩下的缺少装啥就行了

 

1):no module named cv2

pip install opencv-python

2):no module named albumentations

pip install albumentations

3):no module named tensorboard tensorboardX

pip install tensorboard
pip install tensorboardX
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息