本地安装torch2.3.1,cuda12.1,python3.10

卸载之前安装的torch (torch2.2.1+cpu)

pip uninstall torch

pip uninstall torch

下载对应的torch版本及相关文件

torch-2.3.1+cu121-cp310
torchvision-0.18.1+cu121
torchaudio-2.3.1+cu121

下载地址:pytorch-wheels-cu121安装包下载_开源镜像站-阿里云 (aliyun.com)

或百度网盘下载:
链接: https://pan.baidu.com/s/1n0ijikRUSaCCyk310J-b9g?pwd=pa3m 提取码: pa3m 
 

分别安装

pip install "torch-2.3.1+cu121-cp310-cp310-win_amd64.whl"
pip install "torchaudio-2.3.1+cu121-cp310-cp310-win_amd64.whl"
pip install "torchvision-0.18.1+cu121-cp310-cp310-win_amd64.whl"

检查是否安装成功,进入python后输入

 import torch
 print(torch.__version__)
 print(torch.cuda.is_available())

第一个打印 2.3.1+cu121,第二个打印True

参考

使用阿里云镜像本地安装torch的cu121版本示例_torch 2.3.1+cu121-CSDN博客

(vlmgame) hxj@hxj:~/VLN-game/Grounded-Segment-Anything$ pip install -e GroundingDINO Obtaining file:///home/hxj/VLN-game/Grounded-Segment-Anything/GroundingDINO Preparing metadata (setup.py) ... done Requirement already satisfied: torch in /home/hxj/anaconda3/envs/vlmgame/lib/python3.10/site-packages (from groundingdino==0.1.0) (2.1.0) Requirement already satisfied: torchvision in /home/hxj/anaconda3/envs/vlmgame/lib/python3.10/site-packages (from groundingdino==0.1.0) (0.16.0) Requirement already satisfied: transformers in /home/hxj/anaconda3/envs/vlmgame/lib/python3.10/site-packages (from groundingdino==0.1.0) (4.55.4) Requirement already satisfied: addict in /home/hxj/anaconda3/envs/vlmgame/lib/python3.10/site-packages (from groundingdino==0.1.0) (2.4.0) Requirement already satisfied: yapf in /home/hxj/anaconda3/envs/vlmgame/lib/python3.10/site-packages (from groundingdino==0.1.0) (0.40.2) Requirement already satisfied: timm in /home/hxj/anaconda3/envs/vlmgame/lib/python3.10/site-packages (from groundingdino==0.1.0) (1.0.19) Requirement already satisfied: numpy in /home/hxj/anaconda3/envs/vlmgame/lib/python3.10/site-packages (from groundingdino==0.1.0) (2.1.3) Requirement already satisfied: opencv-python in /home/hxj/anaconda3/envs/vlmgame/lib/python3.10/site-packages (from groundingdino==0.1.0) (4.12.0.88) Requirement already satisfied: supervision in /home/hxj/anaconda3/envs/vlmgame/lib/python3.10/site-packages (from groundingdino==0.1.0) (0.26.1) Requirement already satisfied: pycocotools in /home/hxj/anaconda3/envs/vlmgame/lib/python3.10/site-packages (from groundingdino==0.1.0) (2.0.10) Requirement already satisfied: scipy>=1.10.0 in /home/hxj/anaconda3/envs/vlmgame/lib/python3.10/site-packages (from supervision->groundingdino==0.1.0) (1.15.3) Requirement already satisfied: matplotlib>=3.6.0 in /home/hxj/anaconda3/envs/vlmgame/lib/python3.10/site-packages (from supervision->groundingdino==0.1.0) (3.10.0) Requirement already satisfied: pyyaml>=5.3 in /home/hxj/anaconda3/envs/vlmgame/lib/python3.10/site-packages (from supervision->groundingdino==0.1.0) (6.0.2) Requirement already satisfied: defusedxml>=0.7.1 in /home/hxj/anaconda3/envs/vlmgame/lib/python3.10/site-packages (from supervision->groundingdino==0.1.0) (0.7.1) Requirement already satisfied: pillow>=9.4 in /home/hxj/anaconda3/envs/vlmgame/lib/python3.10/site-packages (from supervision->groundingdino==0.1.0) (11.1.0) Requirement already satisfied: requests>=2.26.0 in /home/hxj/anaconda3/envs/vlmgame/lib/python3.10/site-packages (from supervision->groundingdino==0.1.0) (2.32.3) Requirement already satisfied: tqdm>=4.62.3 in /home/hxj/anaconda3/envs/vlmgame/lib/python3.10/site-packages (from supervision->groundingdino==0.1.0) (4.67.1) Requirement already satisfied: contourpy>=1.0.1 in /home/hxj/anaconda3/envs/vlmgame/lib/python3.10/site-packages (from matplotlib>=3.6.0->supervision->groundingdino==0.1.0) (1.3.1) Requirement already satisfied: cycler>=0.10 in /home/hxj/anaconda3/envs/vlmgame/lib/python3.10/site-packages (from matplotlib>=3.6.0->supervision->groundingdino==0.1.0) (0.11.0) Requirement already satisfied: fonttools>=4.22.0 in /home/hxj/anaconda3/envs/vlmgame/lib/python3.10/site-packages (from matplotlib>=3.6.0->supervision->groundingdino==0.1.0) (4.55.3) Requirement already satisfied: kiwisolver>=1.3.1 in /home/hxj/anaconda3/envs/vlmgame/lib/python3.10/site-packages (from matplotlib>=3.6.0->supervision->groundingdino==0.1.0) (1.4.8) Requirement already satisfied: packaging>=20.0 in /home/hxj/anaconda3/envs/vlmgame/lib/python3.10/site-packages (from matplotlib>=3.6.0->supervision->groundingdino==0.1.0) (24.2) Requirement already satisfied: pyparsing>=2.3.1 in /home/hxj/anaconda3/envs/vlmgame/lib/python3.10/site-packages (from matplotlib>=3.6.0->supervision->groundingdino==0.1.0) (3.2.0) Requirement already satisfied: python-dateutil>=2.7 in /home/hxj/anaconda3/envs/vlmgame/lib/python3.10/site-packages (from matplotlib>=3.6.0->supervision->groundingdino==0.1.0) (2.9.0.post0) Requirement already satisfied: six>=1.5 in /home/hxj/anaconda3/envs/vlmgame/lib/python3.10/site-packages (from python-dateutil>=2.7->matplotlib>=3.6.0->supervision->groundingdino==0.1.0) (1.17.0) Requirement already satisfied: charset-normalizer<4,>=2 in /home/hxj/.local/lib/python3.10/site-packages (from requests>=2.26.0->supervision->groundingdino==0.1.0) (2.1.1) Requirement already satisfied: idna<4,>=2.5 in /home/hxj/anaconda3/envs/vlmgame/lib/python3.10/site-packages (from requests>=2.26.0->supervision->groundingdino==0.1.0) (3.7) Requirement already satisfied: urllib3<3,>=1.21.1 in /home/hxj/anaconda3/envs/vlmgame/lib/python3.10/site-packages (from requests>=2.26.0->supervision->groundingdino==0.1.0) (2.3.0) Requirement already satisfied: certifi>=2017.4.17 in /home/hxj/anaconda3/envs/vlmgame/lib/python3.10/site-packages (from requests>=2.26.0->supervision->groundingdino==0.1.0) (2025.4.26) Requirement already satisfied: huggingface_hub in /home/hxj/anaconda3/envs/vlmgame/lib/python3.10/site-packages (from timm->groundingdino==0.1.0) (0.34.4) Requirement already satisfied: safetensors in /home/hxj/anaconda3/envs/vlmgame/lib/python3.10/site-packages (from timm->groundingdino==0.1.0) (0.6.2) Requirement already satisfied: filelock in /home/hxj/anaconda3/envs/vlmgame/lib/python3.10/site-packages (from huggingface_hub->timm->groundingdino==0.1.0) (3.17.0) Requirement already satisfied: fsspec>=2023.5.0 in /home/hxj/anaconda3/envs/vlmgame/lib/python3.10/site-packages (from huggingface_hub->timm->groundingdino==0.1.0) (2025.3.2) Requirement already satisfied: typing-extensions>=3.7.4.3 in /home/hxj/.local/lib/python3.10/site-packages (from huggingface_hub->timm->groundingdino==0.1.0) (4.14.1) Requirement already satisfied: hf-xet<2.0.0,>=1.1.3 in /home/hxj/anaconda3/envs/vlmgame/lib/python3.10/site-packages (from huggingface_hub->timm->groundingdino==0.1.0) (1.1.8) Requirement already satisfied: sympy in /home/hxj/anaconda3/envs/vlmgame/lib/python3.10/site-packages (from torch->groundingdino==0.1.0) (1.13.3) Requirement already satisfied: networkx in /home/hxj/anaconda3/envs/vlmgame/lib/python3.10/site-packages (from torch->groundingdino==0.1.0) (3.4.2) Requirement already satisfied: jinja2 in /home/hxj/anaconda3/envs/vlmgame/lib/python3.10/site-packages (from torch->groundingdino==0.1.0) (3.1.6) Requirement already satisfied: nvidia-cuda-nvrtc-cu12==12.1.105 in /home/hxj/anaconda3/envs/vlmgame/lib/python3.10/site-packages (from torch->groundingdino==0.1.0) (12.1.105) Requirement already satisfied: nvidia-cuda-runtime-cu12==12.1.105 in /home/hxj/anaconda3/envs/vlmgame/lib/python3.10/site-packages (from torch->groundingdino==0.1.0) (12.1.105) Requirement already satisfied: nvidia-cuda-cupti-cu12==12.1.105 in /home/hxj/anaconda3/envs/vlmgame/lib/python3.10/site-packages (from torch->groundingdino==0.1.0) (12.1.105) Requirement already satisfied: nvidia-cudnn-cu12==8.9.2.26 in /home/hxj/anaconda3/envs/vlmgame/lib/python3.10/site-packages (from torch->groundingdino==0.1.0) (8.9.2.26) Requirement already satisfied: nvidia-cublas-cu12==12.1.3.1 in /home/hxj/anaconda3/envs/vlmgame/lib/python3.10/site-packages (from torch->groundingdino==0.1.0) (12.1.3.1) Requirement already satisfied: nvidia-cufft-cu12==11.0.2.54 in /home/hxj/anaconda3/envs/vlmgame/lib/python3.10/site-packages (from torch->groundingdino==0.1.0) (11.0.2.54) Requirement already satisfied: nvidia-curand-cu12==10.3.2.106 in /home/hxj/anaconda3/envs/vlmgame/lib/python3.10/site-packages (from torch->groundingdino==0.1.0) (10.3.2.106) Requirement already satisfied: nvidia-cusolver-cu12==11.4.5.107 in /home/hxj/anaconda3/envs/vlmgame/lib/python3.10/site-packages (from torch->groundingdino==0.1.0) (11.4.5.107) Requirement already satisfied: nvidia-cusparse-cu12==12.1.0.106 in /home/hxj/anaconda3/envs/vlmgame/lib/python3.10/site-packages (from torch->groundingdino==0.1.0) (12.1.0.106) Requirement already satisfied: nvidia-nccl-cu12==2.18.1 in /home/hxj/anaconda3/envs/vlmgame/lib/python3.10/site-packages (from torch->groundingdino==0.1.0) (2.18.1) Requirement already satisfied: nvidia-nvtx-cu12==12.1.105 in /home/hxj/anaconda3/envs/vlmgame/lib/python3.10/site-packages (from torch->groundingdino==0.1.0) (12.1.105) Requirement already satisfied: triton==2.1.0 in /home/hxj/anaconda3/envs/vlmgame/lib/python3.10/site-packages (from torch->groundingdino==0.1.0) (2.1.0) Requirement already satisfied: nvidia-nvjitlink-cu12 in /home/hxj/anaconda3/envs/vlmgame/lib/python3.10/site-packages (from nvidia-cusolver-cu12==11.4.5.107->torch->groundingdino==0.1.0) (12.9.86) Requirement already satisfied: MarkupSafe>=2.0 in /home/hxj/anaconda3/envs/vlmgame/lib/python3.10/site-packages (from jinja2->torch->groundingdino==0.1.0) (3.0.2) Requirement already satisfied: mpmath<1.4,>=1.1.0 in /home/hxj/anaconda3/envs/vlmgame/lib/python3.10/site-packages (from sympy->torch->groundingdino==0.1.0) (1.3.0) Requirement already satisfied: regex!=2019.12.17 in /home/hxj/anaconda3/envs/vlmgame/lib/python3.10/site-packages (from transformers->groundingdino==0.1.0) (2024.11.6) Requirement already satisfied: tokenizers<0.22,>=0.21 in /home/hxj/anaconda3/envs/vlmgame/lib/python3.10/site-packages (from transformers->groundingdino==0.1.0) (0.21.4) Requirement already satisfied: importlib-metadata>=6.6.0 in /home/hxj/anaconda3/envs/vlmgame/lib/python3.10/site-packages (from yapf->groundingdino==0.1.0) (8.5.0) Requirement already satisfied: platformdirs>=3.5.1 in /home/hxj/anaconda3/envs/vlmgame/lib/python3.10/site-packages (from yapf->groundingdino==0.1.0) (4.3.7) Requirement already satisfied: tomli>=2.0.1 in /home/hxj/anaconda3/envs/vlmgame/lib/python3.10/site-packages (from yapf->groundingdino==0.1.0) (2.0.1) Requirement already satisfied: zipp>=3.20 in /home/hxj/anaconda3/envs/vlmgame/lib/python3.10/site-packages (from importlib-metadata>=6.6.0->yapf->groundingdino==0.1.0) (3.21.0) Installing collected packages: groundingdino DEPRECATION: Legacy editable install of groundingdino==0.1.0 from file:///home/hxj/VLN-game/Grounded-Segment-Anything/GroundingDINO (setup.py develop) is deprecated. pip 25.3 will enforce this behaviour change. A possible replacement is to add a pyproject.toml or enable --use-pep517, and use setuptools >= 64. If the resulting installation is not behaving as expected, try using --config-settings editable_mode=compat. Please consult the setuptools documentation for more information. Discussion can be found at https://github.com/pypa/pip/issues/11457 Running setup.py develop for groundingdino error: subprocess-exited-with-error × python setup.py develop did not run successfully. │ exit code: 1 ╰─> [65 lines of output] A module that was compiled using NumPy 1.x cannot be run in NumPy 2.1.3 as it may crash. To support both 1.x and 2.x versions of NumPy, modules must be compiled with NumPy 2.0. Some module may need to rebuild instead e.g. with &#39;pybind11>=2.12&#39;. If you are a user of the module, the easiest solution will be to downgrade to &#39;numpy<2&#39; or try to upgrade the affected module. We expect that some modules will need time to support NumPy 2. Traceback (most recent call last): File "<string>", line 2, in <module> File "<pip-setuptools-caller>", line 35, in <module> File "/home/hxj/VLN-game/Grounded-Segment-Anything/GroundingDINO/setup.py", line 27, in <module> import torch File "/home/hxj/anaconda3/envs/vlmgame/lib/python3.10/site-packages/torch/__init__.py", line 1382, in <module> from .functional import * # noqa: F403 File "/home/hxj/anaconda3/envs/vlmgame/lib/python3.10/site-packages/torch/functional.py", line 7, in <module> import torch.nn.functional as F File "/home/hxj/anaconda3/envs/vlmgame/lib/python3.10/site-packages/torch/nn/__init__.py", line 1, in <module> from .modules import * # noqa: F403 File "/home/hxj/anaconda3/envs/vlmgame/lib/python3.10/site-packages/torch/nn/modules/__init__.py", line 35, in <module> from .transformer import TransformerEncoder, TransformerDecoder, \ File "/home/hxj/anaconda3/envs/vlmgame/lib/python3.10/site-packages/torch/nn/modules/transformer.py", line 20, in <module> device: torch.device = torch.device(torch._C._get_default_device()), # torch.device(&#39;cpu&#39;), /home/hxj/anaconda3/envs/vlmgame/lib/python3.10/site-packages/torch/nn/modules/transformer.py:20: UserWarning: Failed to initialize NumPy: _ARRAY_API not found (Triggered internally at ../torch/csrc/utils/tensor_numpy.cpp:84.) device: torch.device = torch.device(torch._C._get_default_device()), # torch.device(&#39;cpu&#39;), Building wheel groundingdino-0.1.0 Compiling with CUDA running develop /home/hxj/anaconda3/envs/vlmgame/lib/python3.10/site-packages/setuptools/command/develop.py:42: EasyInstallDeprecationWarning: easy_install command is deprecated. !! ******************************************************************************** Please avoid running ``setup.py`` and ``easy_install``. Instead, use pypa/build, pypa/installer or other standards-based tools. See https://github.com/pypa/setuptools/issues/917 for details. ******************************************************************************** !! easy_install.initialize_options(self) /home/hxj/anaconda3/envs/vlmgame/lib/python3.10/site-packages/setuptools/_distutils/cmd.py:66: SetuptoolsDeprecationWarning: setup.py install is deprecated. !! ******************************************************************************** Please avoid running ``setup.py`` directly. Instead, use pypa/build, pypa/installer or other standards-based tools. See https://blog.ganssle.io/articles/2021/10/setup-py-deprecated.html for details. ******************************************************************************** !! self.initialize_options() running egg_info writing groundingdino.egg-info/PKG-INFO writing dependency_links to groundingdino.egg-info/dependency_links.txt writing requirements to groundingdino.egg-info/requires.txt writing top-level names to groundingdino.egg-info/top_level.txt reading manifest file &#39;groundingdino.egg-info/SOURCES.txt&#39; adding license file &#39;LICENSE&#39; writing manifest file &#39;groundingdino.egg-info/SOURCES.txt&#39; running build_ext error: [Errno 2] No such file or directory: &#39;/home/hxj/anaconda3/envs/vlmgame/bin/nvcc&#39; [end of output] note: This error originates from a subprocess, and is likely not a problem with pip. error: subprocess-exited-with-error × python setup.py develop did not run successfully. │ exit code: 1 ╰─> [65 lines of output] A module that was compiled using NumPy 1.x cannot be run in NumPy 2.1.3 as it may crash. To support both 1.x and 2.x versions of NumPy, modules must be compiled with NumPy 2.0. Some module may need to rebuild instead e.g. with &#39;pybind11>=2.12&#39;. If you are a user of the module, the easiest solution will be to downgrade to &#39;numpy<2&#39; or try to upgrade the affected module. We expect that some modules will need time to support NumPy 2. Traceback (most recent call last): File "<string>", line 2, in <module> File "<pip-setuptools-caller>", line 35, in <module> File "/home/hxj/VLN-game/Grounded-Segment-Anything/GroundingDINO/setup.py", line 27, in <module> import torch File "/home/hxj/anaconda3/envs/vlmgame/lib/python3.10/site-packages/torch/__init__.py", line 1382, in <module> from .functional import * # noqa: F403 File "/home/hxj/anaconda3/envs/vlmgame/lib/python3.10/site-packages/torch/functional.py", line 7, in <module> import torch.nn.functional as F File "/home/hxj/anaconda3/envs/vlmgame/lib/python3.10/site-packages/torch/nn/__init__.py", line 1, in <module> from .modules import * # noqa: F403 File "/home/hxj/anaconda3/envs/vlmgame/lib/python3.10/site-packages/torch/nn/modules/__init__.py", line 35, in <module> from .transformer import TransformerEncoder, TransformerDecoder, \ File "/home/hxj/anaconda3/envs/vlmgame/lib/python3.10/site-packages/torch/nn/modules/transformer.py", line 20, in <module> device: torch.device = torch.device(torch._C._get_default_device()), # torch.device(&#39;cpu&#39;), /home/hxj/anaconda3/envs/vlmgame/lib/python3.10/site-packages/torch/nn/modules/transformer.py:20: UserWarning: Failed to initialize NumPy: _ARRAY_API not found (Triggered internally at ../torch/csrc/utils/tensor_numpy.cpp:84.) device: torch.device = torch.device(torch._C._get_default_device()), # torch.device(&#39;cpu&#39;), Building wheel groundingdino-0.1.0 Compiling with CUDA running develop /home/hxj/anaconda3/envs/vlmgame/lib/python3.10/site-packages/setuptools/command/develop.py:42: EasyInstallDeprecationWarning: easy_install command is deprecated. !! ******************************************************************************** Please avoid running ``setup.py`` and ``easy_install``. Instead, use pypa/build, pypa/installer or other standards-based tools. See https://github.com/pypa/setuptools/issues/917 for details. ******************************************************************************** !! easy_install.initialize_options(self) /home/hxj/anaconda3/envs/vlmgame/lib/python3.10/site-packages/setuptools/_distutils/cmd.py:66: SetuptoolsDeprecationWarning: setup.py install is deprecated. !! ******************************************************************************** Please avoid running ``setup.py`` directly. Instead, use pypa/build, pypa/installer or other standards-based tools. See https://blog.ganssle.io/articles/2021/10/setup-py-deprecated.html for details. ******************************************************************************** !! self.initialize_options() running egg_info writing groundingdino.egg-info/PKG-INFO writing dependency_links to groundingdino.egg-info/dependency_links.txt writing requirements to groundingdino.egg-info/requires.txt writing top-level names to groundingdino.egg-info/top_level.txt reading manifest file &#39;groundingdino.egg-info/SOURCES.txt&#39; adding license file &#39;LICENSE&#39; writing manifest file &#39;groundingdino.egg-info/SOURCES.txt&#39; running build_ext error: [Errno 2] No such file or directory: &#39;/home/hxj/anaconda3/envs/vlmgame/bin/nvcc&#39; [end of output] note: This error originates from a subprocess, and is likely not a problem with pip.
最新发布
08-25
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值