2019-04-09
2019-04-27 本文已影响0人
崔雅鑫
anaconda安装pytorch,tensorflow-gpu环境。
前提:笔记本英伟达显卡驱动已经安装好了。
打开anaconda prompt
输入代码
conda create -n conda_env python=3.6
conda activate conda_env
以上两行代码为我个人创建的虚拟环境,你可以不同输入,创建pytorch, tensorflow-gpu环境代码如下,anaconda会为我们自动安装cuda, cudnn
conda install pytorch torchvision cudatoolkit=10.0 -c pytorch
conda install tensorflow-gpu