Windows+GTX1660TI下安装CUDA 10.1&cudnn 7.6.5 + tensorflow/pytorch
CUDA+cudnn
用anaconda安装虚拟环境,会有安装包cudatoolkit10.1,这相当于cuda10.1的一个API接口,cudnn同样如此。所以我们先安装CUDA 10.1&cudnn 7.6.5 ,这个是要看GPU适合哪个版本的,具体看链接: 安装CUDA.
虚拟环境配置
首先要改为国内镜像源链接
conda更换国内源
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --set show_channel_urls yes
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --set show_channel_urls yes
换回默认源
conda config --remove-key channels
pip更改国内源
pip config set global.index-url ht