WIN10+tensorflow-gpu-1.15+torch安装踩过的坑
之前写代码都用的是pytorch,但是对比实验需要对比别人tensorflow的代码,于是需要安装tensorflow,但安装时踩了不少的坑,把我踩过的坑记录一下,也希望能帮到大家!
第一个坑:tensorflow-gpu-1.15需要python3.5-3.7版本
目前大多数tensorflow的代码都是基于tensorflow 1.X编写的,安装tensorflow 2.X需要自己修改很多地方,为了方便起见,直接安装tensorflow-gpu-1.15。
我之前用的是python3.9,发现没法直接安装。尝试过新建一个conda环境,但是安装指定版本的tensorflow总出现以下问题,很头疼。
于是直接安装python3.7得了,配坏境就只需要花时间,总不至于卡住。
至此,使用pip安装了tensorflow-gpu-1.15。
pip就是神!
第二个坑:pytoch需要考虑CUDA+python版本的兼容性
之前装的CUDA11.8,使用pip安装pytorch时发现报错,查了之后发现,CUDA11.8仅支持python3.8及以上的版本
于是又重新安装了CUDA11.7,因为我之前也踩过坑,写过博客,这个也很快搞定。