最近装了自己笔记本的tensorflow-gpu和pytorch,把用到的命令放到这里,写的比较简要
安装教程:
先下载安装anaconda,关于这个有很多教程
打开anaconda prompt
conda create -n cuda10.0 python=3.7 这是创建指定python版本的虚拟环境名称,python=3.7版本下的名为cuda10.0的虚拟环境
conda activate cuda10.0 激活该虚拟环境
conda install pytorch torchvision cudatoolkit=10.0 -c pytorch 到pytorch官网找所需的版本
conda install tensorflow-gpu 或 conda install tensorflow-gpu==1.12.0 不指定就默认安装最新版,也可以指定固定版本,cuda和cudnn会在下载tensorflow过程中自动下载,无需手动配置,以前我都是手动下载好再按网上教程配置。
conda install anaconda 给该虚拟环境中装所需的库
完成!!!
下载安装pycharm,创建新项目,编译器选该环境下的python即可,网上也有很多教程参考。