安装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 创建虚拟环境运行,避免损坏本地环境

猜你喜欢

转载自blog.csdn.net/wuhchengfei616/article/details/114277161