Ubuntu22.04.4系统/安装python3.9/pytorch/torchvision【GPU版】

1.安装python3.9

1.1 创建python3.9的虚拟环境

conda create -n QwenChat python=3.9

1.2 输入"y"

1.3 创建成功

2.安装pytorch和torchvision

2.1 进入虚拟环境

进入刚刚创建的虚拟环境

conda activate QwenChat

2.2 conda安装

  • 查看cuda的版本
  • 浏览器打开网址PyTorch
  • 鼠标往下翻,到参数选择界面(因无CUDA12.2,选择了最接近的CUDA12.1),复制【Run this Command】的命令

conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia

  • 运行【Run this Command】的命令
  • 输入"y"后,稍等片刻,安装完成

2.3 测试

python -c "import torch; print(torch.version)"

python -c "import torchvision; print(torchvision.version)"

相关推荐
程序猿追1 分钟前
探索 CANN Graph 引擎的计算图编译优化策略:深度技术解读
人工智能·目标跟踪
哈__1 分钟前
CANN加速语音识别ASR推理:声学模型与语言模型融合优化
人工智能·语言模型·语音识别
慢半拍iii12 分钟前
CANN算子开发实战:手把手教你基于ops-nn仓库编写Broadcast广播算子
人工智能·计算机网络·ai
历程里程碑19 分钟前
普通数组----合并区间
java·数据结构·python·算法·leetcode·职场和发展·tornado
weixin_3954489120 分钟前
mult_yolov5_post_copy.c_cursor_0205
c语言·python·yolo
User_芊芊君子25 分钟前
CANN数学计算基石ops-math深度解析:高性能科学计算与AI模型加速的核心引擎
人工智能·深度学习·神经网络·ai
小白|28 分钟前
CANN与联邦学习融合:构建隐私安全的分布式AI推理与训练系统
人工智能·机器学习·自动驾驶
艾莉丝努力练剑35 分钟前
hixl vs NCCL:昇腾生态通信库的独特优势分析
运维·c++·人工智能·cann
执风挽^36 分钟前
Python基础编程题2
开发语言·python·算法·visual studio code
梦帮科技36 分钟前
Node.js配置生成器CLI工具开发实战
前端·人工智能·windows·前端框架·node.js·json