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