0
点赞
收藏
分享

微信扫一扫

pytorch-gpu环境配置

_阿瑶 2022-04-25 阅读 77

pytorch-gpu环境配置

1、在线安装
(已配置CUDA11.3,Cudnn8.0,Anaconda新建环境Python=3.9)
进入官网在这里插入图片描述

2、离线安装
首先查询需要安装的各版本对应情况
在这里插入图片描述
在这里插入图片描述
我安装的是torch = 1.10.0,torchvision= 0.11.1, torchaudio = 0.10.0
下载对应的文件(注意如果要安装gpu版本的pytorch,请选择标注了cuda版本的文件并保证cuda与torch的版本对应)下载地址

其中cu113表示对应的是cuda11.3版本

下载完成后将文件放在对应的 lib/site-packages文件夹下,开始安装:

在这里插入图片描述

验证安装是否成功

//进入Python3.9环境
import torch
torch.__version__  # 查看torch版本
torch.cuda.is_available()  # 查看gpu是否可用(返回true表示pytorch-gpu安装成功)

若输出结果如下,则安装成功:

在这里插入图片描述

举报

相关推荐

0 条评论