配置: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