1. 打开anaconda prompt命令行,创建环境

conda create -n torch python=3.7

依次输入y即可

2. 激活环境,命令行输入如下代码,进入torch环境

conda activate torch

3. 进入pytorch官网,选择合适的版本

注意,要和自己电脑的cuda版本一致

如何查看自己电脑的cuda版本:Win+R,cmd打开命令行,输入nvcc -V,查看版本号

如果“找不到命令”,则说明电脑没有CUDA,需要手动下载,参考博客

复制命令,在anaconda prompt里运行

conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch

4. 测试,命令行输入python,输入import torch,不报错说明安装成功。

【深度学习】Anaconda配置Pytorch环境_逍遥鱼的博客-CSDN博客

Logo

一站式 AI 云服务平台

更多推荐