PWC--NET PYTHON3.6 PYTORCH0.2.0环境配置

# setup environment
conda create -n pwcnet_test python=3.6 anaconda
conda activate pwcnet_test
# install pytorch and other dependencies
pip install http://download.pytorch.org/whl/cu80/torch-0.2.0.post2-cp36-cp36m-manylinux1_x86_64.whl
pip install torchvision visdom dominate opencv-python cffi
pip install http://download.pytorch.org/whl/cu80/torch-0.2.0.post2-cp36-cp36m-manylinux1_x86_64.whl
# install external packages 
cd external_packages/correlation-pytorch-master/
bash make_cuda.sh
cd ../../

annaconda/envs/pwcnet-test/lib/python3.6/conv

#30
#33
int(out_channels), int(a//b)

run:

python script_pwc.py './data/frame_0010.png' './data/frame_0011.png' './tmp/frame_0010.flo'

你可能感兴趣的:(PWC--NET PYTHON3.6 PYTORCH0.2.0环境配置)