Windows10深度学习环境配置

2023-02-25

字数统计: 204字 | 阅读时长: 1分

前言

Windows10深度学习环境配置，cuda11.6。

anaconda
cudatoolkit
cudnn
pytorch
tensorflow

Anaconda

anaconda

Cudatoolkit

11.6

全程默认安装，cmd输入nvcc -V检验是否按照成功。

可能需要的环境变量

path

卸载

如正常应用程序卸载即可，只留下：NVIDIA的图形驱动程序、NVIDIA Physx系统软件、NVIDIA GeForce Experience。

cuDNN

https://developer.nvidia.cn/rdp/cudnn-archive

需要注册账户，选择与cudatoolkit对应的cuDNN版本进行下载。

下载下来后，把各文件夹的内容放到CUDA对应目录下。

Pytorch

对于cudatoolkit==11.6，按照官网教程安装。

1	conda install pytorch torchvision torchaudio pytorch-cuda=11.6 -c pytorch -c nvidia

检验：torch.cuda.isavailable()

Tensorflow

python -m pip install tensorflow==2.10.0，2.10.0为最后一个支持Windows调用GPU的版本。

检验：tf.config.list_physical_devices('GPU')

后记

首发于 silencezheng.top，转载请注明出处。

jsonContent: meta: false pages: false posts: title: true date: true path: true text: false raw: false content: false slug: false updated: false comments: false link: false permalink: false excerpt: false categories: false tags: true