尧图建网站 尧图建网站 YAOTU WEB BUILD 免费咨询
ARTICLE DETAIL

资讯详情

深耕网站建设与建站编程的一线实战洞察。

如何安装Pytorch框架简洁教程

如何安装Pytorch框架简洁教程 文章目录Pytorch 框架步骤1步骤2步骤3创建虚拟环境选择合适的下载链接安装验证Pytorch 框架PyTorch 本身是一个 Python 深度学习框架里面有深度学习所用的 Python 包torch torchvision torchaudio。步骤1更新 NVIDIA 最新驱动。步骤2查看 CUDA UMD Version。打开终端输入nvidia-smi注意这里图片中的 CUDA UMD Version 是 13.3步骤3选择下载链接的时候要用到步骤3创建虚拟环境最好在电脑上创建1个虚拟环境将 Pytorch 的包安装在创建的虚拟环境里方便管理。电脑的基础环境名是 base我这里用 conda 创建一个名为“my_env”的虚拟环境并指定这个虚拟环境的 python 版本python 3.11现在很稳定。安装完 conda 之后在终端输入conda create-nmy_envpython3.11选择合适的下载链接进入 Pytorch 官网 https://pytorch.org/往下滑找到这个页面从上往下依次选择一般选择稳定版本Stable但是 5060 显卡应该选用Preview版本。选择你的电脑系统。选择Pip安装。选择Python。注意这里选择的 CUDA 计算平台的版本要低于 上面查看的CUDA UMD Version。复制 Run this Command 后面的安装命令pip3 install --pre torch torchvision --index-url https://......。安装在终端激活你创建的虚拟环境“my_env”终端里输入以下命令激活环境conda activate my_env退出 conda 虚拟环境的命令conda deactivate激活虚拟环境后粘贴上面的安装命令回车运行。pip3install--pretorch torchvision --index-url https://......验证验证是否安装成功在刚才创建的虚拟环境中终端里输入以下命令查看安装的包是否有torch(my_env)conda list终端里输入 python 回车之后输入(my_env)python(my_env)importtorch(my_env)torch.cuda.is_available()True(my_env)exit()显示True证明安装成功。
返回列表