一、安装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)调用torch进行检验
(3)如果有结果则下载成功
在终端输入:
python
import torch
pr

最低0.47元/天 解锁文章
1万+

被折叠的 条评论
为什么被折叠?



