我在anaconda环境中安装了cudatoolkit和cudnn包,但tensorflow不识别我的GPU设备。在一些网络资源中,我看到你可以通过安装必要的anaconda软件包来使用Cuda。所以,我想知道的是,我是否需要从NVIDIA Cuda的网站上安装它才能使用它,还是会导致冲突?
我觉得你的环境变量有问题。我可以推荐以下步骤:
-
创建新环境
-
Install required packages:
conda install -c conda-forge cudatoolkit=11.2 cudnn=8.1.0 python3 -m pip install tensorflow
-
验证安装,首先导入tensorflow
python3 -c "import tensorflow as tf;
-
然后检查GPU
print(tf.config.list_physical_devices('GPU'))"
这应该是工作'