一、安装pytorch12.1
1、安装
(1)打开浏览器,进入pytorch官网选择合适的版本,复制最下面的代码。
(2)打开终端,创建虚拟环境
conda create -n torch12.1cu12.2
(3)激活并进入虚拟环境
conda activate torch12.1cu12.2
(4)将刚才的代码粘贴进行安装
conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch-c nvidia(以官网为准)
2、检验
(1)在终端中打开python
(2&