首先电脑要安装好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了!