其他分享
首页 > 其他分享> > 记录一下我这台电脑搭建pytorch的环境

记录一下我这台电脑搭建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

参考:

标签:1.7,python,torch,环境,pytorch,conda,这台,搭建
来源: https://www.cnblogs.com/wonho/p/15770938.html