简洁安装配置在Windows环境下使用vscode开发pytorch

简洁安装配置在Windows环境下使用vscode开发pytorch

使用anaconda安装pytorch,通过vscode集成环境开发pytorch

下载 anaconda

  1. 下载网址,选择对应系统的版本 https://repo.anaconda.com/archive/
    windows可以选择Anaconda3-2024.10-1-Windows-x86_64.exe
  2. 检查 conda 是否安装成功
    conda --version
  3. 升级 conda
    conda update conda
    添加镜像源
bash 复制代码
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/bioconda/

查看已经添加的channels

bash 复制代码
conda config --get channels
  1. 安装conda完成 配置安装Python虚拟环境
bash 复制代码
conda create --name py311 python=3.11
conda activate py311
  1. 使用命令检查配置环境
bash 复制代码
conda env list 

安装pytorch conda install pytorch

  1. 直接执行
    安装pytorch,它默认安装的是cpu版本的pytorch。
bash 复制代码
conda install pytorch 

要安装很多插件,需要等一段时间

配置vs code 的pytorch的开发环境

  • 安装如下插件

  1. 测试pytorch是否正确安装

    新建一个扩展名为py的文件,代码如下:

    bash 复制代码
    import torch
    
    # 检查是否可以使用GPU
    device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
    print(f"Using device: {device}")
    
    # 创建一个简单的张量
    x = torch.tensor([1.0, 2.0, 3.0], device=device)
    print(f"Tensor x: {x}")
    
    # 创建y张量
    y = torch.tensor([4.0, 5.0, 6.0], device=device)
    print(f"Tensor : {y}")
    # 张量加法
    z = x + y
    print(f"Tensor z (x + y): {z}")
    
    # 张量乘法
    w = x * y
    print(f"Tensor w (x * y): {w}")
  2. 运行py文件,右下角选择py311环境

  3. 输出如下使用CPU的环境

bash 复制代码
Using device: cpu
Tensor x: tensor([1., 2., 3.])
Tensor : tensor([4., 5., 6.])
Tensor z (x + y): tensor([5., 7., 9.])
Tensor w (x * y): tensor([ 4., 10., 18.])
相关推荐
Bellafu6661 天前
selenium常用的等待有哪些?
python·selenium·测试工具
小白学大数据1 天前
Python爬虫常见陷阱:Ajax动态生成内容的URL去重与数据拼接
爬虫·python·ajax
2401_841495641 天前
【计算机视觉】基于复杂环境下的车牌识别
人工智能·python·算法·计算机视觉·去噪·车牌识别·字符识别
Adorable老犀牛1 天前
阿里云-ECS实例信息统计并发送统计报告到企业微信
python·阿里云·云计算·企业微信
倔强青铜三1 天前
苦练Python第66天:文件操作终极武器!shutil模块完全指南
人工智能·python·面试
倔强青铜三1 天前
苦练Python第65天:CPU密集型任务救星!多进程multiprocessing模块实战解析,攻破GIL限制!
人工智能·python·面试
Panda__Panda1 天前
docker项目打包演示项目(数字排序服务)
运维·javascript·python·docker·容器·c#
Lris-KK1 天前
力扣Hot100--94.二叉树的中序遍历、144.二叉树的前序遍历、145.二叉树的后序遍历
python·算法·leetcode
zy_destiny1 天前
【工业场景】用YOLOv8实现抽烟识别
人工智能·python·算法·yolo·机器学习·计算机视觉·目标跟踪
(●—●)橘子……1 天前
记力扣2009:使数组连续的最少操作数 练习理解
数据结构·python·算法·leetcode