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

上一篇下一篇

猜你喜欢

热点阅读