conda配置安装pytorch tensorflow-gpu

Posted leoking01

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了conda配置安装pytorch tensorflow-gpu相关的知识,希望对你有一定的参考价值。

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

 

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的主要内容,如果未能解决你的问题,请参考以下文章