安装CUDA cudnn pytorch

安装 python3.6 和 python包

python3.6：https://www.python.org/ftp/python/3.6.7/python-3.6.7.exe

pytorch：https://pytorch.org/ 

pip install torch==1.7.1+cu101 torchvision==0.8.2+cu101 torchaudio===0.7.2 -f https://download.pytorch.org/whl/torch_stable.html

安装cuda10.1，cudnn

cuda10.1: https://developer.nvidia.com/cuda-toolkit-archive

cudnn: https://developer.nvidia.com/rdp/cudnn-download

 

使用 virtualenv 创建虚拟环境运行，避免损坏本地环境