Win10系统下配置深度学习环境

之前在本机配置深度学习环境时候踩了很多坑,这里稍作记录备忘。

Anaconda

安装Anaconda

在anaconda官网下载安装。

安装完成后,1. 配置环境变量;2. 修改anaconda的安装源

验证conda:

conda -V
conda env list

创建conda环境

# 创建一个名称为common的虚拟环境并指定python版本为3.6.8
conda create -n common python=3.6.8
# 激活虚拟环境
activate common
# 列出所有的虚拟环境
conda env list
# 退出虚拟环境
conda deactivate
# 删除虚拟环境
conda remove -n common --all

安装NVIDIA显卡驱动

官网选择电脑相应的显卡版本进行下载:https://www.nvidia.cn/geforce/drivers/

安装CUDA

nvidia官网下载对应版本:

https://developer.nvidia.cn/cuda-toolkit-archive

安装完毕后,1. 添加环境变量(如果安装时自动添加则跳过这一步)2. 检查CUDA是否安装成功

nvcc -V

安装CUDNN

NVIDIA cuDNN是用于深度神经网络的GPU加速库。

官网地址:https://developer.nvidia.com/rdp/cudnn-archive

选择和CUDA版本相对应的进行下载。

下载万恒后,把解压得到的文件夹内的bin、include、lib目录下的dll文件与h文件分别复制到相应的CUDA的安装目录下。

通过NVIDIA提供的 deviceQuery.exebandwidthTest.exe 来查看GPU的状态,两者均在安装目录的 extras\demo_suite文件夹中

安装Pytorch

https://pytorch.org/get-started/previous-versions/ 中选择与CUDA版本对应的Pytorch版本,执行给出的命令进行安装。

这里推荐pip安装,在尝试的时候发现使用conda命令固然可以安装,但是安装之后torch.cuda.is_available()为False

后来选择pip安装,报错,后选择Pytorch低一点的版本,成功torch.cuda.is_available()==True

安装完成后,进入Python命令行,执行

import torch
print(torch.cuda.is_available())

结果为True则torch-GPU可用。

注意事项

  1. 从显卡型号到显卡驱动,到CUDA,到CUDNN,到Pytorch,到torchvision,到cudatoolkit这些版本一定要对应好。
  2. 安装pytorch选择命令时,选择pip手动下载whl文件安装较保险。

参考资料

  1. 给win11装个深度学习环境
  2. 为何torch.version.cuda返回None?
  3. pytorch gpu不可用 原因核查
  4. 显卡、显卡驱动、cuda 之间的关系是什么?
文章作者: Met Guo
文章链接: https://guoyujian.github.io/2022/01/16/Win10%E7%B3%BB%E7%BB%9F%E4%B8%8B%E9%85%8D%E7%BD%AE%E6%B7%B1%E5%BA%A6%E5%AD%A6%E4%B9%A0%E7%8E%AF%E5%A2%83/
版权声明: 本博客所有文章除特别声明外,均采用 CC BY-NC-SA 4.0 许可协议。转载请注明来自 Gmet's Blog