配置pytorch环境

最新推荐文章于 2024-08-20 09:37:32 发布

nil000000

最新推荐文章于 2024-08-20 09:37:32 发布

阅读量608

点赞数 2

文章标签： pytorch 人工智能 python 深度学习

本文链接：https://blog.csdn.net/qq_45857546/article/details/131504420

版权

该文详细介绍了如何在AnacondaPrompt下创建新的虚拟环境PytorchEnv，使用Python3.8版本。接着，文章指导读者如何激活此环境并安装CUDA工具包11.3，以及特定版本的PyTorch（1.10.2）、torchvision和torchaudio，强调需注意选择GPU版本而非CPU版本。最后，说明了在PyCharm中如何关联此虚拟环境来创建新项目。

摘要由CSDN通过智能技术生成

文章目录

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

配置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下载版本）