3060显卡搭建Pytorch历程

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