国内源安装pytorch gpu版本(cuda 11.1)

部署运行你感兴趣的模型镜像

说明

官网安装完全可以用,只是有点慢。

方法

官网源3个G下载要好几个小时,而且国内源没有11.1的,这里可以用11.3的

pip install torch==1.11.0+cu113 torchvision==0.12.0+cu113 torchaudio==0.11.0 -f https://mirrors.aliyun.com/pytorch-wheels/cu113

您可能感兴趣的与本文相关的镜像

PyTorch 2.5

PyTorch 2.5

PyTorch
Cuda

PyTorch 是一个开源的 Python 机器学习库,基于 Torch 库,底层由 C++ 实现,应用于人工智能领域,如计算机视觉和自然语言处理

为了找到与 PyTorch 1.9.0 和 CUDA 11.1 兼容的 `pytorch3d` 版本,需要考虑以下几个方面: ### PyTorchCUDA 的兼容性 PyTorch 1.9.0 是一个支持 CUDA 11.1版本。在安装时,通常会自动选择与当前 CUDA 工具包兼容的 PyTorch 构建版本。因此,如果系统中使用的是 CUDA 11.1,则 PyTorch 1.9.0 应该能够正常运行[^1]。 ### pytorch3d 的版本要求 `pytorch3d` 是由 Facebook AI Research 维护的一个用于 3D 深度学习的库,其依赖于特定版本PyTorch,并且对 CUDA 支持有特定的要求。对于 PyTorch 1.9.0 和 CUDA 11.1 来说,推荐使用的 `pytorch3d` 版本是 **v0.6.2**。 此版本经过测试,确保了与 PyTorch 1.9.0 的兼容性,并且在构建过程中可以正确链接到 CUDA 11.1 提供的库。 ### 安装建议 可以通过以下方式安装兼容的版本: ```bash pip install pytorch3d==0.6.2 ``` 或者从 GitHub 源代码安装以确保兼容性: ```bash git clone https://github.com/facebookresearch/pytorch3d.git cd pytorch3d git checkout v0.6.2 pip install -e . ``` ### 验证安装 安装完成后,可以通过运行验证脚本来确认是否成功安装并能够在 GPU 上运行: ```python import torch import pytorch3d def test_pytorch3d(): print("PyTorch3D imported successfully!") if torch.cuda.is_available(): device = torch.device("cuda") print("CUDA is available. Testing PyTorch3D with GPU...") # 创建一个 tensor 并移动到 GPU x = torch.tensor([[1.0, 2.0, 3.0]], device=device) # 确保点是二维的 from pytorch3d.transforms import RotateAxisAngle rot = RotateAxisAngle(angle=90, axis="X").to(device) # 将变换移动到 GPU x_transformed = rot.transform_points(x) print("Tensor on GPU: ", x) print("Transformed Tensor: ", x_transformed) else: print("CUDA is not available. Please check your PyTorch and GPU settings.") test_pytorch3d() ``` 上述代码将检查 `pytorch3d` 是否能够正确导入并在 GPU 上执行基本操作[^2]。 ### 总结 - **PyTorch 版本**: 1.9.0 - **CUDA 版本**: 11.1 - **推荐 pytorch3d 版本**: 0.6.2 如果用户希望进一步确认环境中的软件版本,可以运行以下命令来检查 PyTorch 和 torchvision 的版本: ```bash python -c "import torch; print('PyTorch version:', torch.__version__)" python -c "import torchvision; print('torchvision version:', torchvision.__version__)" ``` 这有助于确保所有组件都处于正确的版本状态[^3]。 ---
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值