Anconda 安装pytorch 步骤

首先电脑要安装好Anconda程序,安装好之后,启动下边的命令行。

从电脑中打开Anconda命令行,输入下边的命令创建环境,其中python版本可以根据需要的适配版本进行选择。下图是pytorch对应的版本与python对应的版本号,下表来自pytorch的github官方文档:https://github.com/pytorch/vision#installation

本文是安装的python 3.10版本的环境,命令如下:

conda create -n torchgpu python=3.10

静等安装结束之后,激活该环境,命令如下:

conda activate torchgpu

随后我们登入官网去下载pytorch:Pytorch

打开后可以在这里选择我们要安装的系统,安装方式,可以选择pip安装或者是Conda,如果Conda安装不上就试用另一种方式。

如果电脑没有显卡,我们就安装CPU版本,在图中选择Cpu的安装命令即可。

如果我们需要安装的是GPU版本,我们需要提前确定我们的CUDA号,打开我们电脑自身的命令提示符,输入如下图指令,查看我们的Cuda版本号,我的CUDA号为12.3:

因为我电脑可以使用GPU版本,所以我安装CUDA版本,从pytorch得到相应的安装命令:

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121

静等安装即可,看到下边就可认为安装成功

安装成功,我们可以通过命令:

pip list

查看本环境下我们安装的库,如果我们的torch,torchaudio,torchvision,后边的版本号都带有+cu,说明我们安装的GPU版本成功,如果安装的是CPU版本,则版本号上没有cu。

GPU版本:

CPU版本:

至此就可以使用我们的pytorch了!

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值