浏览 59
扫码
安装PyTorch可以通过多种方式进行,下面我将为您提供两种常用的安装方法:
- 使用conda安装(推荐):
首先,确保您已经安装了Anaconda或者Miniconda。如果没有安装,可以在官方网站下载安装:https://www.anaconda.com/products/distribution
接下来,打开一个终端窗口(Windows系统可以使用Anaconda Prompt),输入以下命令,创建一个新的conda环境并安装PyTorch:
conda create -n pytorch_env python=3.8
conda activate pytorch_env
conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch -c nvidia
这将安装PyTorch和相关的库,包括torchvision和torchaudio。如果您的电脑支持CUDA,还会安装对应的CUDA工具包。安装完成后,可以使用以下命令验证PyTorch是否安装成功:
python -c "import torch; print(torch.__version__)"
- 使用pip安装:
如果您没有使用Anaconda,也可以使用pip来安装PyTorch。在终端窗口中输入以下命令:
pip install torch torchvision torchaudio
这将通过pip安装PyTorch和相关的库。同样,您可以使用以下命令验证PyTorch是否安装成功:
python -c "import torch; print(torch.__version__)"
无论您选择哪种方法安装PyTorch,都可以在Python环境中导入PyTorch并开始使用了。希望这个教程对您有帮助!