✨想要搭建深度学习环境却卡在了安装PyTorch和TorchVision上?别担心,这篇文章帮你轻松搞定!首先,确保你的电脑已安装好Python和pip。接着,打开终端或命令行工具,输入以下命令:
```bash
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
```
💻如果你有NVIDIA显卡并想利用GPU加速,记得将`cpu`替换为`cu118`(对应CUDA 11.8)。安装完成后,可以通过以下代码验证是否成功:
```python
import torch
print(torch.__version__)
print(torch.cuda.is_available())
```
✅如果显示版本号且`cuda.is_available()`为True,说明安装成功啦!✨
💡小贴士:建议安装与显卡驱动匹配的CUDA版本,否则可能会遇到兼容性问题哦。如果还有疑问,可以参考官方文档进一步排查问题。💪
🌟 PyTorch和TorchVision是深度学习入门的绝佳组合,快去试试吧!🔥
免责声明:本文由用户上传,如有侵权请联系删除!