在Ubuntu系统上获取PyTorch资源有多种途径,以下是一些常用的方法:
使用Anaconda来安装PyTorch是一个常见的选择,因为它可以简化环境管理和包的安装过程。可以通过以下命令在Ubuntu上使用Anaconda安装PyTorch:
wget https://repo.anaconda.com/archive/Anaconda3-2023.07-2-Linux-x86_64.sh
bash Anaconda3-2023.07-2-Linux-x86_64.sh
conda create -n pytorch_env python=3.8
conda activate pytorch_env
conda install pytorch torchvision torchaudio cudatoolkit=11.7 -c pytorch -c nvidia
如果你不想使用Anaconda,可以直接使用pip来安装PyTorch。以下是在Ubuntu上使用pip安装PyTorch的命令:
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
pip3 install torch torchvision torchaudio
jupyter
和nb_conda
包:conda install jupyter nb_conda -y
import torch
print(torch.__version__)
print(torch.cuda.is_available())
以上就是在Ubuntu系统上获取和安装PyTorch资源的途径。希望这些信息能帮助你顺利搭建起深度学习环境。如果在安装过程中遇到问题,可以参考相关的教程或在社区论坛寻求帮助。