1、利用conda创建虚拟环境
首先下载安装anaconda,然后在anaconda prompt中输入下列代码,分别创建pytorch和tensorflow的虚拟环境:
#创建pytorch_gpu环境
conda create -n pytorch_gpu python=3.7
#pytorch_gpu为环境名,python=3.7是指python版本为3.7
#创建tensorflw_gpu环境
conda create -n tf2_gpu python=3.7
#pytorch_gpu为环境名,python=3.7是指python版本为3.7
2、安装pytorch的GPU版本
由于在国内通过pytorch官网下载pytorch很慢而导致无法下载完成进而安装失败,所以,我们将使用清华镜像,代码如下:
conda config --add channels https://mirrors.ustc.edu.cn/anaconda/cloud/pytorc

本文详细介绍了如何使用Anaconda创建PyTorch和TensorFlow的GPU环境,包括环境配置、软件安装及验证过程,特别针对国内用户提供了清华镜像加速下载的方法。

1976

被折叠的 条评论
为什么被折叠?



