Win10装tensorflow2.0gpu和pytorch

  • 简介
  • 一、安装Anaconda
  • 二、安装Anaconda-gpu版
  • 1.试过很多种错误,推荐这种傻瓜式安装(如果想在虚拟环境种安装请新建一个虚拟环境,我直接在base下安装的)
  • 2.升级为tensorflow2.0
  • 2.安装cudacnn 和cudatoolkit
  • 三、 pytorch和torchvision


简介

心态炸裂。。。学到torchvision的时候,发现怎么也装不上了,python版本不对,想着重新换个虚拟环境,结果nb_conda也安装不上,花了一下午,干脆重来重来全都重来,重装Anconda,原来的python基础也不要了。终于成功了,介绍下流程。


提示:以下是本篇文章正文内容,下面案例可供参考

一、安装Anaconda

安装这个很简单,附送一个清华Anaconda镜像源·国外的真的是太慢了啊啊啊啊啊啊
链接: link.

二、安装Anaconda-gpu版

1.试过很多种错误,推荐这种傻瓜式安装(如果想在虚拟环境种安装请新建一个虚拟环境,我直接在base下安装的)

pytorch和tensorflow和tansform pytorch和tensorflow2_pytorch


先选uninstalled,里面找到tensorflow安装,安装以后的版本应该是1.2(反正很低)。这时候我们在Anaconda prompt中,使用,conda info tensorflow 命令,看升级它需要的依赖。

2.升级为tensorflow2.0

pytorch和tensorflow和tansform pytorch和tensorflow2_anaconda_02


能看到依赖的包以及每个包的版本,回到Anaconda Navigator,找到相应的这几个包(一个一个的)升级到相应版本。最后再升级Tensorflow2.0。

(别问我为啥不直接安2.0,试了好多次,都失败了,这个方法至少能看到每一步的进度)

注意:
1:依赖的包都升级到2.0版本就可以,虽然有的说>2.0都可以,但是后期还是会出现各种警告。
2:安装后使用jupyter,import tensorflow,出现警告FutureWarning: Conversion of the second argument of issubdtype from float to np.floating is。是h5py包版本的问题,升级到2.8.0就好了。
以上所有东西都已经安装成功。

2.安装cudacnn 和cudatoolkit

这个我直接再Conda prompt中用命令:
conda install cudatoolkit=10.0 cudnn=7
一次成功。(这里选10.0后面pytorch选1.6就好)

三、 pytorch和torchvision

这个的方法和上面tensorflow安装的其实都一样。而且感觉更简单一些。参考上面就好。

第一篇blog,如有不懂的地方,可留言