参考文章:https://blog.csdn.net/m0_49090516/article/details/113576003
我自己的显卡为RTX3060:window安装的最新的英伟达驱动
在创建的虚拟环境只要不超过window的驱动就可以。
自己安装的是cudatoolkit=11.3.1 cudnn=8.2.1 tensorflow-gpu=2.5.0(安装用的手机热点+vpn)
与前文参考博主不同,我安装tensorflow时不能使用pip安装,然后用来conda成功了,最后测试gpu也可以使用。(我的虚拟环境创建的是python3.7版本)
安装pytorch时安装cudatoolkit包记得关vpn,不然很慢。(p站小土堆)