【配置】tensorflow.config.list_physical_devices(“GPU“) []是空

配置:cuda11.2,cuDNN8.1.0,tensorflow2.7.0,pytorch1.7.1

  • 问题:tensorflow.config.list_physical_devices("GPU") []是空

        解决:将cuDNN的lib64和include下的文件都copy到cuda相应的位置下

                cp cuda/lib64/* /usr/local/cuda-11.2/lib64/

                cp cuda/include/* /usr/local/cuda-11.2/include/

  • 问题:torch.cuda.is_availble() False

        解决:cuda11.2没有对应的pytorch,pytorch和cudatoolkit版本不对应,可以使用以下的命令

conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cudatoolkit=11.0 -c pytorch -c conda-forge

你可能感兴趣的:(环境配置,tensorflow)