Pytorch基本使用

1. 基本介绍

        pytorch 是一个开源机器学习和深度学习框架,允许使用Python代码操作和处理数据并编写深度学习算法，能够在强大的GPU加速基础上实现张量和动态神经网络。

2.PyTorch

        · 创建一个虚拟环境

conda create -n xx_name python=3.9

conda activate xx_name

# 查看虚拟环境列表

conda info --envs
conda info -e
conda env list

# 激活虚拟环境
conda activate xx_name

# 删除
conda rename -n xx_name --all

        安装验证

        

import  torch

#Pytorch 和cuda 版本
torch.__version__
torch.version.cuda

# 测试GPU是否生效
torch.cuda.is_available()