Windows安装PyTorch

文章目录

前言

此份文档适用于Windows安装PyTorch,参看此份文档安装PyTorch的前提是之前已参照y以下视频安装好了conda环境。

链接: https://pan.baidu.com/s/1pWVPMc2vysfEPxLRCV_UNw?pwd=3pda 提取码: 3pda

MacOS安装过程见链接

CPU版本安装

安装步骤

  1. 新建一个conda环境
bash 复制代码
conda create -n torch_39 python=3.9
conda activate torch_39


  1. 用pip命令安装torch
bash 复制代码
pip install -i https://pypi.tuna.tsinghua.edu.cn/simple torch torchvision torchaudio
  1. Jupyter Lab虚拟环境的配置

这一步是要将此conda环境"torch_39",添加进Jupyter Lab的Kernel

bash 复制代码
//安装ipykernel
conda install ipykernel 

//在ipykernel中安装当前环境
python -m ipykernel install --name torch_39

conda deactivate

此时打开Jupyter Lab切换Kernel,已出现刚刚安装的"torch_39"conda环境。

测试CPU版PyTorch是否安装成功

GPU版本安装

新建一个conda环境

bash 复制代码
conda create -n torch_GPU python=3.9
conda activate torch_GPU

注:

这里使用的实验室测试老机器的cuda版本为11.1,显卡为730

安装torch

法一:在线安装(建议用法二)

安装11.0版本的:

bash 复制代码
pip install torch==1.7.1+cu110 torchvision==0.8.2+cu110 torchaudio===0.10.1 -f https://download.pytorch.org/whl/torch_stable.html

法二:下载torch、torchvision后本地安装(建议使用此方法)

特别注意torch、torchvision、torchaudio版本对应关系

此处看一下参考文章1中的"五"

参考文章2:PyTorch中torch、torchvision、torchaudio版本对应关系

如果下载地慢,这里挂vpn下载:


切换到文件夹,使用命令安装:

bash 复制代码
pip install torch-1.8.1+cu111-cp39-cp39-win_amd64.whl

pip install ...

Jupyter Lab虚拟环境的配置

这一步是要将此conda环境"torch_GPU",添加进Jupyter Lab的Kernel

bash 复制代码
//安装ipykernel
conda install ipykernel 

//在ipykernel中安装当前环境
python -m ipykernel install --name torch_GPU

conda deactivate

此时打开Jupyter Lab切换Kernel,已出现刚刚安装的"torch_GPU"conda环境。

测试是否安装成功

python 复制代码
import torch

# 验证PyTorch是否安装成功
print("PyTorch version:", torch.__version__)

# 验证CUDA是否可用
print("CUDA available:", torch.cuda.is_available())

# 如果CUDA可用,打印CUDA版本和设备信息
if torch.cuda.is_available():
    print("CUDA version:", torch.version.cuda)
    print("CUDA device name:", torch.cuda.get_device_name())

运行截图:

相关推荐
Jamie2019010610 分钟前
健康孪生智能体使用起来复杂吗?医者AI技术核心与用户体验
人工智能
GLAB-Mary15 分钟前
AI会取代网络工程师吗?理解AI在网络安全中的角色
网络·人工智能·web安全
道可云23 分钟前
道可云人工智能每日资讯|浦东启动人工智能创新应用竞赛
人工智能·百度·ar·xr·deepseek
kyle~27 分钟前
目标检测在国防和政府的应用实例
人工智能·目标检测·计算机视觉
兮℡檬,38 分钟前
torchvision中的数据使用
人工智能
Qdgr_1 小时前
价值实证:数字化转型标杆案例深度解析
大数据·数据库·人工智能
c++服务器开发1 小时前
一文详解Character AI:实用指南+ ChatGPT、Gemini对比分析
人工智能·chatgpt
hanniuniu131 小时前
AI时代API挑战加剧,API安全厂商F5护航企业数字未来
人工智能·安全
李洋-蛟龙腾飞公司1 小时前
HarmonyOS NEXT应用元服务常见列表操作分组吸顶场景
linux·运维·windows
nicepainkiller2 小时前
anchor 智能合约案例3 之 journal
人工智能·智能合约·solana·anchor