首页 > 其他 > 详细

conda配置安装pytorch tensorflow-gpu

时间:2020-07-07 09:40:44      阅读:82      评论:0      收藏:0      [点我收藏+]

//--------------------------------------

 

1、新建环境,避免与基础环境出现冲突

conda create -n pytorch_gpu python=3.7

2、切换到pytorch环境

conda activate pytorch_gpu

3、安装基本框架

conda install pandas
conda install jupyter notebook
......
conda install pytorch torchvision cudatoolkit=10.1

 

 

 

//-------------------------------------

利用conda安装指定版本的tensorflow  

 conda install --channel https://conda.anaconda.org/anaconda tensorflow=1.6.0

 

conda install pytorch torchvision cudatoolkit=10.1 -c pytorch

conda install --channel https://conda.anaconda.org/anaconda tensorflow=1.8.0



//------------------------------------------------------
conda install --channel https://conda.anaconda.org/jjh_cio_testing tensorflow-gpu







conda配置安装pytorch tensorflow-gpu

原文:https://www.cnblogs.com/leoking01/p/13258723.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!