Pytorch环境部署

一、创建虚拟环境

envc文件夹主要存放你所创建的环境

首先打开Anaconda Prompt；

创建虚拟环境：conda create -n torch python=3.8

创建成功后，输入命令： conda env list  查看torch环境已经创建，星号为当前所在环境;

二、安装pytorch

切换到torch环境：conda activate torch

 进入官网（PyTorch）查看要下载的版本：

conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch

将上面代码复制到 pytorch环境终端即可完成环境部署

三、验证pytorch是否安装成功

直接输入 python，即可成功进入

 输入如下指令，查看torch是否安装成功

>>> import torch
>>> x=torch.randn(4)
>>> print(x)

四、cuda安装

NVDIA控制面板–>帮助–>系统信息；查看电脑支持cuda版本

 CUDA Toolkit 11.7 Update 1 Downloads | NVIDIA Developer下载所对应版本，安装都是默认选项