美文网首页
2019-04-09

2019-04-09

作者: 崔雅鑫 | 来源:发表于2019-04-27 23:45 被阅读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

相关文章

网友评论

      本文标题:2019-04-09

      本文链接:https://www.haomeiwen.com/subject/mxmiiqtx.html