cuda_env


cuda环境安装

一、 不同版本的tensorflow-gpu与CUDA的对应关系

老版本的对应关系:

1

较新版本的对应关系:

2

cuda下载地址:https://developer.nvidia.com/cuda-toolkit-archive

cuDNN下载地址: https://developer.nvidia.com/rdp/cudnn-archive

cuda版本查看: nvcc -V

二、安装cuda后的环境变量配置

需要配置的环境变量有:

CUDA_SDK_PATH:C:\ProgramData\NVIDIA Corporation\CUDA Samples\v10.1
CUDA_LIB_PATH:%CUDA_PATH%\lib\x64
CUDA_BIN_PATH:%CUDA_PATH%\bin
CUDA_SDK_BIN_PATH:%CUDA_SDK_PATH%\bin\win64
CUDA_SDK_LIB_PATH:%CUDA_SDK_PATH%\common\lib\x64

同时需要在系统变量的path中配置:

%CUDA_LIB_PATH%
%CUDA_BIN_PATH%
%CUDA_SDK_LIB_PATH%
%CUDA_SDK_BIN_PATH%

3

三、安装anaconda并构建一个python3.6的环境(不装直接用本地环境也可)

anaconda下载地址:https://www.anaconda.com/

构建环境:

conda create -n gpu python=3.6.5
activate gpu
pip install tensorflow==2.6.2 -i https://pypi.tuna.tsinghua.edu.cn/simple
pip install tensorflow-gpu==2.6.2 -i https://pypi.tuna.tsinghua.edu.cn/simple

测试gpu是否可用:

import tensorflow as tf
tf.test.is_gpu_available()
tf.config.list_physical_devices('GPU')

4

如果出现缺库的情况,注意检查环境变量、tensorflow版本和python版本是否对应


文章作者: cfrost
版权声明: 本博客所有文章除特別声明外,均采用 CC BY 4.0 许可协议。转载请注明来源 cfrost !
  目录