1.安装Anaconda, 安装在D盘
3.官网下载Cudnn-11.0和CUDA_11.0,注意版本号对齐
4.安装CUDA,点击cuda_11.exe文件,选择自定义安装
5.安装完后,打开cudn根目录,即C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.0
6.把Cudnn里面的内容复制到该目录下
7.打开命令指示符,输入conda create -n pytorch python=3.6 创建环境
#或者输入conda create -n your_env_name python=3.6 anaconda
#这样创建的环境可以有anaconda的基础科学包,如果不用可以去掉最后面的anaconda
#最好是用3.6版本的python
#环境会自动装在D:\anaconda\envs\PyTorch
8.命令提示符,输入conda activate PyTorch, 进入PyTorch环境
9.输入 pip install torch==1.8.0 torchvision==0.9.0 torchaudio==0.8.0
#3060的显卡安装1.8版本的torch
11.验证安装是否成功 输入python 输入import torch 退出exit()
12.pycharm把环境改为类似D:\anaconda\envs\pytorch\python.exe