配置环境:Win10、anaconda
在CMD中输入:
创建环境
conda create -n pytorch python=3.6
激活环境
activate pytorch
针对CUDA 10.0
pip install torch===1.2.0 torchvision===0.4.0 -f https://download.pytorch.org/whl/torch_stable.html
更新pip
python -m pip install --upgrade pip
pip install tqdm
pip install cupy-cuda100
pip install matplotlib
pip install opencv-python