【深度学习】不同虚拟环境安装不同Pytorch版本

不同虚拟环境安装不同Pytorch-GPU版本

步骤

在成功安装Anaconda的基础上，执行以下操作：

步骤

step1: 打开Pytorch官网，复制指令

Pytorch官网
选择复制红框内指令，在prompt命令窗口右击以复制，并回车执行
在这里插入图片描述

在这里插入图片描述

step2: 检查cudnn是否安装

此时未安装
若已安装，请直接跳至step4进行测试即可.

但一般情况下都未安装，使用conda search cudnn查询可安装的cudnn版本

在这里插入图片描述

可使用conda search cudnn==8.2.1 --info查询特定版本的cudnn需要的cudatoolkit的版本要求:
在这里插入图片描述
*也可在https://developer.nvidia.com/rdp/cudnn-archive查看，此处不讲。

step3: 选择合适的cudnn版本进行安装

在这里插入图片描述

step4: 测试代码

import torch
print(torch.cuda.current_device())   # 返回当前设备索引
print(torch.cuda.device_count())    # 返回GPU的数量
print(torch.cuda.get_device_name(0))   # 返回gpu名字，设备索引默认从0开始
print(torch.cuda.is_available())   # cuda是否可用
print(torch.version.cuda)  # cuda版本
1
2
3
4
5
6

正常结果类似以下截图：
在这里插入图片描述

相关阅读:
【Unity入门计划】基本概念(8)-瓦片地图 TileMap 02
Trapezoidal Rule Integral
陌生人真的会传授你赚钱的技能吗？
计算机网络概述
AutoSAR EcuM系列04- Flex EcuM的ECU启动阶段详细教程
【JVM基础】Java双亲委派机制
PCL入门（四）：kdtree简单介绍和使用
成像雷达量产突破：木牛携手全球合作伙伴突破智驾瓶颈
day10--232.用栈实现队列+ 225. 用队列实现栈+20. 有效的括号+ 1047. 删除字符串中的所有相邻重复项
pycharm2020无法打开，点击无反应

原文地址：https://blog.csdn.net/two_ye/article/details/126022670