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

下载安装PyTorch

深度学习 带GPU的pytorch 的安装及Conda 的配置

深度学习环境安装配置中各个软件的关系及作用(Anaconda,Pycharm,Python,库,PyTorch, conda)

pytorch入门教程 | 第一章:Tensor

转载conda install pytorch 安装缓慢的解决办法

转载conda install pytorch 安装缓慢的解决办法