1.硬件环境
- Linux(Ubuntu20.04)
- GTX1060 6G显存
2.CUDA+cuDNN+Anaconda
- 安装cuda(推荐安装cuda10.1):https://developer.nvidia.com/cuda-toolkit-archive
- 安装cudnn(推荐安装cuDNN v7.6.0):https://developer.nvidia.com/zh-cn/cudnn
- 安装Anaconda(推荐安装Anaconda3-2020.11-Windows-x86_64.exe):https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/
- 安装CUDA(下一步就行), 安装cuDNN(解压->覆盖文件目录:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1), 安装anaconda(下一步,添加到环境变量)
- 检测是否安装正常:
nvidia-smi
,nvcc -V
,conda -V
,python --version
3.环境准备
3.1虚拟环境的创建
- 添加Anaconda的TUNA镜像:
conda config --