Pytorch开发环境搭建(windows、conda)

环境说明

系统:windows 11

IDE: pycharm

工具:Anaconda

安装conda

到官网下装对应的版本安装即可
https://www.anaconda.com/download

安装后创建一个新的python环境

安装pytorch

按官网指引 https://pytorch.org/get-started/locally/

要先在cmd上检查一下自己的显卡CUDA版本,然后选上面对应的CUDA版本

启动上面创建的环境后,执行下面的pip3命令安装即可

复制代码
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

测试效果

配置IDE

新建一个脚本文件,并执行相关的测试代码

代码如下

复制代码
import torch
print(torch.__version__)
print(torch.cuda.is_available())
相关推荐
柴 基15 分钟前
PyTorch 使用指南
人工智能·pytorch·python
空中湖3 小时前
PyTorch武侠演义 第一卷:初入江湖 第7章:矿洞中的计算禁制
人工智能·pytorch·python
江山如画,佳人北望4 小时前
pytorch常用函数
人工智能·pytorch·python
边缘常驻民1 天前
PyTorch深度学习入门记录3
人工智能·pytorch·深度学习
AndrewHZ1 天前
【图像处理基石】如何对遥感图像进行目标检测?
图像处理·人工智能·pytorch·目标检测·遥感图像·小目标检测·旋转目标检测
墨染点香1 天前
第七章 Pytorch构建模型详解【构建CIFAR10模型结构】
人工智能·pytorch·python
兮℡檬,1 天前
房价预测|Pytorch
人工智能·pytorch·python
贝塔西塔2 天前
PytorchLightning最佳实践基础篇
pytorch·深度学习·lightning·编程框架
小猪和纸箱2 天前
通过Python交互式控制台理解Conv1d的输入输出
pytorch
love530love2 天前
命令行创建 UV 环境及本地化实战演示—— 基于《Python 多版本与开发环境治理架构设计》的最佳实践
开发语言·人工智能·windows·python·conda·uv