anaconda下 tensflow-gpu和torch(gpu)共存搭建

环境信息:

windows10+cuda10.1

踩过的坑:

1.在Anaconda Navigator里面安装的tensorflow-gpu, cudatoolkit会锁定在10.0 ,而pytorch在写这文章的时候就已经是支持10.1的了,如果改成10.1会造成tensorflow-gpu的卸载并安装cpu的版本

2.我在conda增加了清华源安装pytorch(因为快一些),安装后发现怎么也无法使用pytorch的gpu,命令一查,发现根本就没有,也就是说清华源安装的pytorch很有可能是cpu的,看了列表确实是,如果你从清华源安装了后没有办法使用gpu的话,需要从官网进行下载

 

安装步骤:

安装cuda10.1和cudnn7---> 再从官网安装pytorch --> 再使用pip 安装tensorflow-gpu既可

 

1.安装cuda10.1和cudnn7,都可以再英伟达网站找到,目前发布了cuda10.2,暂时不要下载

cuda10.1   https://developer.nvidia.com/cuda-toolkit-archive

cudnn   https://developer.nvidia.com/cudnn

2.从pytorch下载安装

网站: https://pytorch.org/get-started/locally/

conda install pytorch torchvision cudatoolkit=10.1 -c pytorch

-c 和pytorch 是为了从官网下载,如果去掉后,可以从自定义源中下载

3.使用pip或者pip3 安装tensorflow-gpu

pip install tensorflow-gpu