配置pytorch环境

  • Post author:
  • Post category:其他




配置pytorch环境(均在Anaconda Prompt下输入命令)



1.创建新的虚拟环境

conda create -n PytorchEnvs python=3.8 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main



2.进入创建的虚拟环境

conda activate PytorchEnvs



3.下载镜像源中存在的pytorch三个组合版本

conda install cudatoolkit=11.3 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
conda install pytorch=1.10.2 torchvision torchaudio -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/
(指定好pytorch版本,否则会下载成cpu的pytorch,下载时看下下载的是不是gpu,是再点y否则点n确定pytorch下载版本)



4.使用pycharm创建新项目时使用前面创建好的虚拟环境



版权声明:本文为qq_45857546原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。