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())
相关推荐
William.csj16 小时前
Pytorch/CUDA——flash-attn 库编译的 gcc 版本问题
pytorch·cuda
Green1Leaves1 天前
pytorch学习-9.多分类问题
人工智能·pytorch·学习
摸爬滚打李上进2 天前
重生学AI第十六集:线性层nn.Linear
人工智能·pytorch·python·神经网络·机器学习
HuashuiMu花水木2 天前
PyTorch笔记1----------Tensor(张量):基本概念、创建、属性、算数运算
人工智能·pytorch·笔记
阳懿2 天前
【已解决】执行conda init提示No action taken.
conda
try2find2 天前
移动conda虚拟环境的安装目录
linux·运维·conda
喝过期的拉菲2 天前
如何使用 Pytorch Lightning 启用早停机制
pytorch·lightning·早停机制
kk爱闹2 天前
【挑战14天学完python和pytorch】- day01
android·pytorch·python
Yo_Becky2 天前
【PyTorch】PyTorch预训练模型缓存位置迁移,也可拓展应用于其他文件的迁移
人工智能·pytorch·经验分享·笔记·python·程序人生·其他
xinxiangwangzhi_2 天前
pytorch底层原理学习--PyTorch 架构梳理
人工智能·pytorch·架构