简易配置GPU环境下的tensorflow和pytorch

简易配置GPU环境下的tensorflow和pytorch
本机环境:GPU为1660ti,电脑的NVIDIA驱动版本大于cuda10.2所需的最低NVIDIA驱动版本,配置完成实测可以使用
cuda10.2
cudnn与cuda10.2对应版本
电脑的NVIDIA驱动版本≥cuda10.2所需的最低NVIDIA驱动版本
tensorflow安装tensorflow-gpu==2.3.4(复制cudart_102.dll改名为cudart_101.dll)
pytorch安装
pip3 install torch1.10.1+cu102 torchvision0.11.2+cu102 torchaudio===0.10.1+cu102 -f https://download.pytorch.org/whl/cu102/torch_stable.html

你可能感兴趣的:(笔记)