pytorch的安装

pytorch的安装

第一步： 新建虚拟环境

注：在Anaconda Prompt (Anaconda3)内执行如下命令

conda create -n pytorch pip python=3.7

第二步：激活新建的虚拟环境

activate pytorch

第三步： 安装pytorch

conda install pytorch==1.6.0 torchvision==0.7.0 cudatoolkit=9.2 -c pytorch

第四步： 安装tenorflow-cpu(可不安装)

pip install tensorflow-cpu==2.3.1 -i https://pypi.douban.com/simple

OVER！