首先,下载anaconda,如何下载,详见官网
1.在Anaconda Prompt中,
conda create -n 环境名 python=3.8
(环境名建议取自己不会忘记的)
(conda env list 可以显示所以你创建过的虚拟环境)
2.activate 环境名
3.下载一些安装包
(conda list 可以显示该虚拟环境中所以你下载过的安装包)
pip install torch==1.8.0+cu111 torchvision==0.9.0+cu111 torchaudio==0.8.0 -f https://download.pytorch.org/whl/torch_stable.html
4.(我看到其他博客说,如果要求下载cuda和cudnn,install到虚拟环境中就可,记得查找自己适合的版本!!!)
conda install cudatoolkit=11.1 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/win-64/
conda install cudnn=7.6.5 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/win-64/
或者
conda install cudnn=7.6.5
或者自动匹配版本
conda install cudnn
5.打开pycharm,配置你创建的虚拟环境
祝顺利!!!