windows10 python3.7环境安装:
建议创建一个新的python环境来使用pytorch
使用conda安装的命令:
conda install pytorch==1.4.0 torchvision==0.5.0 cudatoolkit=10.1
使用pip3安装的命令:
pip install torch==1.4.0 torchvision==0.5.0
下载cuda和cudnn:
cuda10.1下载网址https://developer.nvidia.com/cuda-10.0-download-archive?target_os=Windo