PyTorch2学习之路1
1.安装PyTorch2,使用conda安装。需要安装cuda或者cpu版本二选一。
安装命令:
conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch-nightly -c nvidia
整体需要安装挺长时间。
2.为了使用方便,需要安装jupyter和huggingface
1.安装PyTorch2,使用conda安装。需要安装cuda或者cpu版本二选一。
安装命令:
conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch-nightly -c nvidia
整体需要安装挺长时间。
2.为了使用方便,需要安装jupyter和huggingface