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下安装的)
先选uninstalled,里面找到tensorflow安装,安装以后的版本应该是1.2(反正很低)。这时候我们在Anaconda prompt中,使用,conda info tensorflow 命令,看升级它需要的依赖。
2.升级为tensorflow2.0
能看到依赖的包以及每个包的版本,回到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,如有不懂的地方,可留言