windows10 基于Ananconda 安装pytorch GPU版本只要输入代码即可（亲测有效）

最新推荐文章于 2024-10-05 12:00:56 发布

方淮哥

最新推荐文章于 2024-10-05 12:00:56 发布

阅读量190

点赞数 2

文章标签： pytorch windows 深度学习

本文链接：https://blog.csdn.net/qq_36693767/article/details/121147721

版权

在开始程序中打开Anaconda Prompt一行一行输入

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/pro/
conda config --set show_channel_urls yes

conda create -n pytorch python=3.8
（注释：通过conda create -n pytorch python=3.9命令创建pytorch环境，python的版本可以自由选择）

conda activate pytorch

conda install pytorch torchvision torchaudio cudatoolkit=10.2

测试一下

import torch

torch.cuda.is_available()