本地深度学习环境配置
1.安装NVIDA驱动,CUDA和CuDNN
通过NVIDA控制面板-系统信息-组件,查看显卡支持的CUDA版本,根据CUDA版本安装相应CuDNN;
驱动:https://www.nvidia.cn/geforce/drivers/
CuDNN:https://developer.nvidia.com/rdp/cudnn-archive#a-collapse742-10
下载好的CuDNN(11.1版本)解压后所有文件夹放到NVIDIA GPU Computing Toolkit\CUDA\v11.1文件夹下。
30系显卡只支持CUDA11,pytorch目前(2021/4/23)支持CUDA11.1
2.conda安装pytorch不稳定(镜像也不稳定),pip快且稳定
Pytorch官网:https://pytorch.org/get-started/locally/
Anaconda Prompt窗口输入命令
pip3 install t