记录一下我这台电脑搭建pytorch的环境

笔记本rtx2060

#创建环境
conda create -n torch python=3.8
#激活环境
conda activate torch
#安装
conda install pytorch=1.7.0 torchvision=0.8.1 torchaudio cudatoolkit=10.2
#查看一下列表
conda list
#进入python编进
python
#导入
>>>import torch
#版本信息
>>> torch.__version__
'1.7.0'
#验证pytorch是否可以使用我的GPU
>>> torch.cuda.is_available()
True

参考:
记录一下我这台电脑搭建pytorch的环境

上一篇:多层感知机fashion mnist


下一篇:YOLOX改进之损失函数(下)