快速在Windows搭建tf和pytorch深度学习环境
不想看文章,看这里
终极解决方案
首先请到tensorflow官方网站查看需要使用的依赖版本
https://www.tensorflow.org/install/source_windows?hl=zh_cn#gpu
在这里找到tensorflow需要使用的cuda
和cudnn
版本
在conda里面输入下面命令
1 | conda install -c conda-forge cudatoolkit=(cuda版本) |
截至2020年十月,该命令没有问题
1 | conda install pytorch torchvision cudatoolkit=10.1 -c pytorch |
前言
目前网上的方法都比较复杂,在看了pytorch和tensorflow的安装文档并踩了很多的坑之后,本人找到了一种全新的方法快速安装环境
准备
在开始之前,你应该先安装好anconda环境
查看版本
首先到官网查看pytorch的版本
再到tf官网查看版本要求
这里有很多可以修改的选项,前四项按照截图选择,最后一项cuda版本需要注意。
我们需要同时安装两个环境,因此目前只能选择cuda10.1版本
安装
此时请复制pytorch(cuda10.1)的安装命令并完成安装。
1 | conda install pytorch torchvision cudatoolkit=10.1 -c pytorch |
接下来是配置tf,输入以下命令可以安装cudnn
1 | conda install -c anaconda cudnn |
此时你不需要关心版本问题,conda会为你安装好正确的版本。如果该命令运行时间过长则需要注意是否不存在该cuda版本对应的cudnn版本
在这之后你就可以使用pip安装tensorflow了
1 | pip install tensorflow |
本博客所有文章除特别声明外,均采用 CC BY-NC-SA 4.0 许可协议。转载请注明来自 RainChan的小博客!
评论