1.Anaconda3安装
- Anaconda Prompt 命令行下创建虚拟环境‘名字’
conda create -n 名字 python=3.6
激活环境 activate 名字
3.查看显卡版本
4.选择安装对应的
cuda 11.1
https://developer.nvidia.com/cuda-11.1.0-download-archive?target_os=Windows&target_arch=x86_64&target_version=10&target_type=exelocal
cuDNNv8.0.5 https://developer.nvidia.com/rdp/cudnn-archive
5.选择对应的Pytorch1.8版本及torchvision0.9版本
https://pytorch.org/get-started/previous-versions/