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())

运行截图:

相关推荐
读创商闻43 分钟前
极狐GitLab CEO 柳钢——极狐 GitLab 打造中国企业专属 AI 编程平台,引领编程新潮流
人工智能·gitlab
kailp43 分钟前
语言模型玩转3D生成:LLaMA-Mesh开源项目
人工智能·3d·ai·语言模型·llama·gpu算力
marteker44 分钟前
弗兰肯斯坦式的人工智能与GTM策略的崩溃
人工智能·搜索引擎
无心水1 小时前
大语言模型零样本情感分析实战:无需机器学习训练,96%准确率实现指南
人工智能·机器学习·语言模型
来自于狂人1 小时前
AI大模型训练的云原生实践:如何用Kubernetes指挥千卡集群?
人工智能·云原生·kubernetes
橡晟6 小时前
深度学习入门:让神经网络变得“深不可测“⚡(二)
人工智能·python·深度学习·机器学习·计算机视觉
墨尘游子6 小时前
神经网络的层与块
人工智能·python·深度学习·机器学习
Leah01056 小时前
什么是神经网络,常用的神经网络,如何训练一个神经网络
人工智能·深度学习·神经网络·ai
PyAIExplorer7 小时前
图像亮度调整的简单实现
人工智能·计算机视觉
Striker_Eureka8 小时前
DiffDet4SAR——首次将扩散模型用于SAR图像目标检测,来自2024 GRSL(ESI高被引1%论文)
人工智能·目标检测