0
点赞
收藏
分享

微信扫一扫

windows安装pytorch

金刚豆 2023-09-21 阅读 48

windows安装pytorch

1. 安装cuda

pytorch官网我要安装1.12.1对应的cuda有三个版本,我选了11.6
在这里插入图片描述
去官网安装这个版本的cuda下载链接
在这里插入图片描述
安装后打开命令行输入nvcc -V,可以显示版本则安装成功,如果显示nvcc不是外部命令,进入安装文件下的bin目录,在当前文件夹打开命令行输入.\nvcc.exe -V 正确显示版本信息
在这里插入图片描述
在这里插入图片描述

2. 安装cudann

安装地址
在这里插入图片描述
选择对应版本安装
解压后把这三个文件复制到之前cuda的安装目录
在这里插入图片描述
在这里插入图片描述
安装完毕

3. 安装pytorch

conda中创建新的环境,conda create -n XXX python=3.7

进入环境

conda中创建新的环境,conda activate XXX

安装pytorch

conda install pytorch==1.12.1 torchvision==0.13.1 torchaudio==0.12.1 cudatoolkit=11.6 -c pytorch -c conda-forge

要是安装出错就换pip

pip install torch==1.12.1+cu116 torchvision==0.13.1+cu116 torchaudio==0.12.1 --extra-index-url https://download.pytorch.org/whl/cu116

测试

命令行输入python进入python环境

import torch
torch.__version__
torch.cuda.is_available()

在这里插入图片描述
安装完成

举报

相关推荐

0 条评论