Windows上搭建Pytorch平台

NosONE

已于 2024-04-12 11:49:25 修改

阅读量524

点赞数

分类专栏：问题解决文章标签： pytorch windows python

于 2022-06-22 15:17:28 首次发布

本文链接：https://blog.csdn.net/amateurSU/article/details/125408146

版权

1 篇文章 0 订阅

订阅专栏

电脑配置详情：RTX3060, cuda 11.4, windows11
在这里插入图片描述

总结：具体步骤如下：

安装anaconda——> torch官网查看对应版本——>安装cuda——>安装cudnn
——>创建环境——>安装GPU版本的pytorch——>验证搭建完成

附加设置：

进入该网站可查看到适配的cuda版本为11.3，且能获得安装指令Run this Command:
conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch
（安装的pytorch版本为1.11.0）

在这里插入图片描述

选择：在这里插入图片描述
点击后按图选取：

cuda和cudnn的安装过程参考此博客，下载版本选择如下图在这里插入图片描述

首先，需要使用清华镜像源，参考该博客。

步骤1：打开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 --set show_channel_urls yes

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/

步骤2：创建pytorch环境，输入下面代码，使用的python版本是3.7.3（其他版本的python应该也可以）：

conda create -n pytorch_env_name python=3.7.3`

步骤3：进入环境

activate pytorch_env_name

步骤4：输入torch的安装指令Run this Command（使用清华源后要删去 -c pytorch）

conda install pytorch torchvision torchaudio cudatoolkit=11.3

步骤5：搭建完成！！！

此处查看，在pycharm中如何使用搭建的pytorch环境。
输入以下代码验证是否搭建成功：

import torch
print("torch {}".format(torch.__version__))
print(torch.cuda.is_available())
print(torch.version.cuda)

成功将显示
在这里插入图片描述

关注

专栏目录