前面有一篇博客专门讲了安装CPU版本的Pytorch,因为当时没有GPU,现在有了3090,专门记录一下安装GPU版的过程。
一、添加清华源
可参考官方anaconda | 镜像站使用帮助 | 清华大学开源软件镜像站 | Tsinghua Open Source Mirror
创建虚拟环境
若没有创建虚拟环境,则需要创建虚拟环境,用来根据实际所需安装、管理各类package
(1)查看当前已存在虚拟环境列表
conda env list
(2)创建新的虚拟环境
conda create --name pytorch-cpu python=3.9 #pytorch-cpu为虚拟环境名称
(3)进入虚拟环境