记录一下安装CUDA和CUDnn的良心帖子
1.总体步骤+事前准备
windows下CUDA的卸载以及安装_花花少年的博客-CSDN博客_卸载cuda
2.CUDA+CUDnn安装具体步骤
CUDA、CUDNN在windows下的安装及配置_花花少年的博客-CSDN博客_cudnn配置
3.torch官网查询:
3.1 记得先添加清华园镜像:
conda添加清华镜像源_eefresher的博客-CSDN博客_conda添加清华镜像
*补充:
- https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
- https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
- https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
- https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
- https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/pro/
3.2 pytorch官网:
3.3TensorFlow-gpu安装:
pip install tensorflow-gpu
4.结果检查:
查看CUDA:
nvcc -V
查看torch能否用gpu:
import torch
torch.cuda.is_available()
查看TensorFlow能否用gpu:
import tensorflow
tensorflow.test.is_gpu_available()
记录完毕~