我尝试在Anaconda中安装TensorFlow v2.12,使用Python 3.9.16,在Windows 10中安装pip v23.0.1**。我的GPU(RTX 4080)需要Tensforflow v2.12,只有这个版本才能与我的GPU配合使用,因为支持Cuda Toolkit v11.8,这是最早支持Ada Lovelace GPU的版本。
Windowstensorflow.org操作系统的官方www.example.com提供了以下安装说明。
conda install -c conda-forge cudatoolkit=11.8.0
python3 -m pip install nvidia-cudnn-cu11==8.6.0.163 tensorflow==2.12.*
CUDNN_PATH=$(dirname $(python -c "import nvidia.cudnn;print(nvidia.cudnn.__file__)"))
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:$CONDA_PREFIX/lib/:$CUDNN_PATH/lib
验证安装:
python3 -c "import tensorflow as tf; print(tf.config.list_physical_devices('GPU'))"
在以上步骤中,Cuda Toolkit 11.8.0和Tensorflow 2.12安装完毕,没有任何问题。但是,如果我尝试cuDNN8.6.0.163安装,没有安装,并给予以下结果。
Looking in indexes: https://pypi.org/simple, https://pypi.ngc.nvidia.com
ERROR: Could not find a version that satisfies the requirement nvidia-cudnn-cu11==8.6.0.163 (from versions: 0.0.1.dev5, 2021.10.26, 2021.11.18, 2021.12.8, 2022.1.13, 2022.4.2, 2022.5.19)
ERROR: No matching distribution found for nvidia-cudnn-cu11==8.6.0.163
出了什么问题?有没有关于如何安装cuDNN www.example.com的想法8.6.0.163?
我尝试用以下方法安装cuDNN8.6.0.163,结果都是一样的:
python -m pip install nvidia-cudnn-cu11==8.6.0.163
(我的conda环境中的python是v3.9.16,所以python3命令不存在,它与python相同。我已经检查了它:python --version
)pip install nvidia-cudnn-cu11==8.6.0.163
python -m pip install nvidia-cudnn-cu11==8.6.0.163 --no-cache-dir
python -m pip install nvidia-cudnn-cu11==8.6.0.163 --proxy="xxx.xxx.xxx.xxx:xxxx"
(其中xxx.xxx.xxx.xxx:xxxx部分代理IP及对应端口)
另外,我安装了'nvidia-pyindex':pip install nvidia-pyindex
,但这并不能解决问题。
在所有这些之后,我尝试8.6.0.163在Windows环境中安装cuDNN www.example.com版本,并且我设置了适当的环境变量,但Anaconda中的Tenrosflow没有看到,等等:print(tf.config.list_physical_devices('GPU'))
不返回任何内容:[ ]
从第3步开始,我检查了以下内容:python -c "import nvidia.cudnn;print(nvidia.cudnn.__file__)
其中该返回:
Traceback (most recent call last):
File "<string>", line 1, in <module>
ModuleNotFoundError: No module named 'nvidia'
在PowerShell中:
nvidia-smi
返回:
“NVIDIA-SMI 531.41|驱动程序版本:531.41|CUDA版本:12.1英寸nvcc -V
返回:
nvcc:NVIDIA(R)Cuda compiler driver Copyright(c)2005-2022 NVIDIA Corporation Built on Wed_Sep_21_10:41:10_Pacific_Daylight_Time_2022 Cuda compilation tools,release 11.8,V11.8.89 Build cuda_11.8.r11.8/compiler.31833905_0
1条答案
按热度按时间omhiaaxx1#
这里的问题是你试图在Windows操作系统中使用
TensorFlow 2.12
支持GPU。但是TensorFlow在本机Win操作系统中的TF 2.10版本之后停止了GPU支持,根据这个doc,它提到:TensorFlow 2.10是在原生Windows上支持GPU的最后一个TensorFlow版本。从TensorFlow 2.11开始,您需要在WSL 2中安装TensorFlow,或者安装tensorflow-cpu,并可选地尝试TensorFlow-DirectML-Plugin
因此,请通过降级
TensorFlow 2.10
并安装兼容的CUDA 11.2
和cuDNN 8.1
(如本测试构建配置中所述)或使用WSL 2或directML插件在您的系统中通过Tensorflow获得GPU支持来重试。