当前位置： 首页 > news >正文

timm==0.5.4 cuda=11.8如何配置环境

news 2025/9/5 17:07:46

timm==0.5.4 cuda=11.8如何配置环境

一种是conda安装，但是一直装的cpu版本的torch

# 1. 创建环境
conda create -n timm_env python=3.9 -y# 2. 激活环境
conda activate timm_env# 3. 一步到位安装所有东西
conda install pytorch==1.13.1 torchvision==0.14.1 torchaudio==0.13.1 pytorch-cuda=11.8 -c pytorch -c nvidia -y# 4. 安装 timm
pip install timm==0.5.4# 5. 验证
python -c "import torch; print(torch.cuda.is_available())"
#False

另一种是pip安装，安装的是gpu版的torch，可以使用

# 1. 创建纯净环境
conda create -n timm_final python=3.9 -y# 2. 激活环境
conda activate timm_final# 3. 使用 Pip 安装 PyTorch GPU 版本
pip install torch==1.13.1+cu117 torchvision==0.14.1+cu117 torchaudio==0.13.1 --extra-index-url https://download.pytorch.org/whl/cu117# 4. 安装 timm
pip install timm==0.5.4# 5. 验证
python -c "import torch; print(torch.cuda.is_available())"
#True

下面是整个过程

E:\AI\Code\OpenSourceCode\G2L-Stereo>conda create -n timm_final python=3.9 -y
Channels:- pytorch- nvidia- defaults
Platform: win-64
Collecting package metadata (repodata.json): done
Solving environment: done## Package Plan ##environment location: D:\ProgramData\anaconda3\envs\timm_finaladded / updated specs:- python=3.9The following packages will be downloaded:package                    |            build---------------------------|-----------------python-3.9.23              |       h716150d_0        16.7 MBsetuptools-78.1.1          |   py39haa95532_0         1.7 MBvc-14.3                    |      h2df5915_10          19 KBvs2015_runtime-14.44.35208 |      ha6b5a95_10          19 KBwheel-0.45.1               |   py39haa95532_0         145 KB------------------------------------------------------------Total:        18.5 MBdone
#
# To activate this environment, use
#
#     $ conda activate timm_final
#
# To deactivate an active environment, use
#
#     $ conda deactivateE:\AI\Code\OpenSourceCode\G2L-Stereo>conda activate timm_final(timm_final) E:\AI\Code\OpenSourceCode\G2L-Stereo>pip install torch==1.13.1+cu117 torchvision==0.14.1+cu117 torchaudio==0.13.1 --extra-index-url https://download.pytorch.org/whl/cu117
Looking in indexes: https://pypi.org/simple, https://download.pytorch.org/whl/cu117
Collecting torch==1.13.1+cu117Downloading https://download.pytorch.org/whl/cu117/torch-1.13.1%2Bcu117-cp39-cp39-win_amd64.whl (2255.6 MB)━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 2.3/2.3 GB 6.7 MB/s  0:03:32
Collecting torchvision==0.14.1+cu117Downloading https://download.pytorch.org/whl/cu117/torchvision-0.14.1%2Bcu117-cp39-cp39-win_amd64.whl (4.8 MB)━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 4.8/4.8 MB 4.7 MB/s  0:00:01
Collecting torchaudio==0.13.1Downloading https://download.pytorch.org/whl/cu117/torchaudio-0.13.1%2Bcu117-cp39-cp39-win_amd64.whl (2.3 MB)━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 2.3/2.3 MB 4.0 MB/s  0:00:00
Collecting typing-extensions (from torch==1.13.1+cu117)Downloading typing_extensions-4.15.0-py3-none-any.whl.metadata (3.3 kB)
Collecting numpy (from torchvision==0.14.1+cu117)Downloading numpy-2.0.2-cp39-cp39-win_amd64.whl.metadata (59 kB)
Collecting requests (from torchvision==0.14.1+cu117)Downloading requests-2.32.5-py3-none-any.whl.metadata (4.9 kB)
Collecting pillow!=8.3.*,>=5.3.0 (from torchvision==0.14.1+cu117)Downloading pillow-11.3.0-cp39-cp39-win_amd64.whl.metadata (9.2 kB)
Collecting charset_normalizer<4,>=2 (from requests->torchvision==0.14.1+cu117)Downloading charset_normalizer-3.4.3-cp39-cp39-win_amd64.whl.metadata (37 kB)
Collecting idna<4,>=2.5 (from requests->torchvision==0.14.1+cu117)Using cached idna-3.10-py3-none-any.whl.metadata (10 kB)
Collecting urllib3<3,>=1.21.1 (from requests->torchvision==0.14.1+cu117)Downloading urllib3-2.5.0-py3-none-any.whl.metadata (6.5 kB)
Collecting certifi>=2017.4.17 (from requests->torchvision==0.14.1+cu117)Downloading certifi-2025.8.3-py3-none-any.whl.metadata (2.4 kB)
Downloading pillow-11.3.0-cp39-cp39-win_amd64.whl (7.0 MB)━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 7.0/7.0 MB 6.2 MB/s  0:00:01
Downloading numpy-2.0.2-cp39-cp39-win_amd64.whl (15.9 MB)━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 15.9/15.9 MB 9.9 MB/s  0:00:01
Downloading requests-2.32.5-py3-none-any.whl (64 kB)
Downloading charset_normalizer-3.4.3-cp39-cp39-win_amd64.whl (107 kB)
Using cached idna-3.10-py3-none-any.whl (70 kB)
Downloading urllib3-2.5.0-py3-none-any.whl (129 kB)
Downloading certifi-2025.8.3-py3-none-any.whl (161 kB)
Downloading typing_extensions-4.15.0-py3-none-any.whl (44 kB)
Installing collected packages: urllib3, typing-extensions, pillow, numpy, idna, charset_normalizer, certifi, torch, requests, torchvision, torchaudio
Successfully installed certifi-2025.8.3 charset_normalizer-3.4.3 idna-3.10 numpy-2.0.2 pillow-11.3.0 requests-2.32.5 torch-1.13.1+cu117 torchaudio-0.13.1+cu117 torchvision-0.14.1+cu117 typing-extensions-4.15.0 urllib3-2.5.0(timm_final) E:\AI\Code\OpenSourceCode\G2L-Stereo>pip install timm==0.5.4
Collecting timm==0.5.4Using cached timm-0.5.4-py3-none-any.whl.metadata (36 kB)
Requirement already satisfied: torch>=1.4 in d:\programdata\anaconda3\envs\timm_final\lib\site-packages (from timm==0.5.4) (1.13.1+cu117)
Requirement already satisfied: torchvision in d:\programdata\anaconda3\envs\timm_final\lib\site-packages (from timm==0.5.4) (0.14.1+cu117)
Requirement already satisfied: typing-extensions in d:\programdata\anaconda3\envs\timm_final\lib\site-packages (from torch>=1.4->timm==0.5.4) (4.15.0)
Requirement already satisfied: numpy in d:\programdata\anaconda3\envs\timm_final\lib\site-packages (from torchvision->timm==0.5.4) (2.0.2)
Requirement already satisfied: requests in d:\programdata\anaconda3\envs\timm_final\lib\site-packages (from torchvision->timm==0.5.4) (2.32.5)
Requirement already satisfied: pillow!=8.3.*,>=5.3.0 in d:\programdata\anaconda3\envs\timm_final\lib\site-packages (from torchvision->timm==0.5.4) (11.3.0)
Requirement already satisfied: charset_normalizer<4,>=2 in d:\programdata\anaconda3\envs\timm_final\lib\site-packages (from requests->torchvision->timm==0.5.4) (3.4.3)
Requirement already satisfied: idna<4,>=2.5 in d:\programdata\anaconda3\envs\timm_final\lib\site-packages (from requests->torchvision->timm==0.5.4) (3.10)
Requirement already satisfied: urllib3<3,>=1.21.1 in d:\programdata\anaconda3\envs\timm_final\lib\site-packages (from requests->torchvision->timm==0.5.4) (2.5.0)
Requirement already satisfied: certifi>=2017.4.17 in d:\programdata\anaconda3\envs\timm_final\lib\site-packages (from requests->torchvision->timm==0.5.4) (2025.8.3)
Using cached timm-0.5.4-py3-none-any.whl (431 kB)
Installing collected packages: timm
Successfully installed timm-0.5.4(timm_final) E:\AI\Code\OpenSourceCode\G2L-Stereo>python -c "import torch; print(torch.cuda.is_available())"
True(timm_final) E:\AI\Code\OpenSourceCode\G2L-Stereo>pip install tqdm
Collecting tqdmUsing cached tqdm-4.67.1-py3-none-any.whl.metadata (57 kB)
Collecting colorama (from tqdm)Using cached colorama-0.4.6-py2.py3-none-any.whl.metadata (17 kB)
Using cached tqdm-4.67.1-py3-none-any.whl (78 kB)
Using cached colorama-0.4.6-py2.py3-none-any.whl (25 kB)
Installing collected packages: colorama, tqdm
Successfully installed colorama-0.4.6 tqdm-4.67.1(timm_final) E:\AI\Code\OpenSourceCode\G2L-Stereo>pip install matplotlib
Collecting matplotlibDownloading matplotlib-3.9.4-cp39-cp39-win_amd64.whl.metadata (11 kB)
Collecting contourpy>=1.0.1 (from matplotlib)Downloading contourpy-1.3.0-cp39-cp39-win_amd64.whl.metadata (5.4 kB)
Collecting cycler>=0.10 (from matplotlib)Using cached cycler-0.12.1-py3-none-any.whl.metadata (3.8 kB)
Collecting fonttools>=4.22.0 (from matplotlib)Downloading fonttools-4.59.2-cp39-cp39-win_amd64.whl.metadata (111 kB)
Collecting kiwisolver>=1.3.1 (from matplotlib)Downloading kiwisolver-1.4.7-cp39-cp39-win_amd64.whl.metadata (6.4 kB)
Requirement already satisfied: numpy>=1.23 in d:\programdata\anaconda3\envs\timm_final\lib\site-packages (from matplotlib) (2.0.2)
Collecting packaging>=20.0 (from matplotlib)Using cached packaging-25.0-py3-none-any.whl.metadata (3.3 kB)
Requirement already satisfied: pillow>=8 in d:\programdata\anaconda3\envs\timm_final\lib\site-packages (from matplotlib) (11.3.0)
Collecting pyparsing>=2.3.1 (from matplotlib)Downloading pyparsing-3.2.3-py3-none-any.whl.metadata (5.0 kB)
Collecting python-dateutil>=2.7 (from matplotlib)Using cached python_dateutil-2.9.0.post0-py2.py3-none-any.whl.metadata (8.4 kB)
Collecting importlib-resources>=3.2.0 (from matplotlib)Using cached importlib_resources-6.5.2-py3-none-any.whl.metadata (3.9 kB)
Collecting zipp>=3.1.0 (from importlib-resources>=3.2.0->matplotlib)Downloading zipp-3.23.0-py3-none-any.whl.metadata (3.6 kB)
Collecting six>=1.5 (from python-dateutil>=2.7->matplotlib)Using cached six-1.17.0-py2.py3-none-any.whl.metadata (1.7 kB)
Downloading matplotlib-3.9.4-cp39-cp39-win_amd64.whl (7.8 MB)━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 7.8/7.8 MB 6.8 MB/s  0:00:01
Downloading contourpy-1.3.0-cp39-cp39-win_amd64.whl (211 kB)
Using cached cycler-0.12.1-py3-none-any.whl (8.3 kB)
Downloading fonttools-4.59.2-cp39-cp39-win_amd64.whl (1.5 MB)━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 1.5/1.5 MB 3.9 MB/s  0:00:00
Using cached importlib_resources-6.5.2-py3-none-any.whl (37 kB)
Downloading kiwisolver-1.4.7-cp39-cp39-win_amd64.whl (55 kB)
Using cached packaging-25.0-py3-none-any.whl (66 kB)
Downloading pyparsing-3.2.3-py3-none-any.whl (111 kB)
Using cached python_dateutil-2.9.0.post0-py2.py3-none-any.whl (229 kB)
Using cached six-1.17.0-py2.py3-none-any.whl (11 kB)
Downloading zipp-3.23.0-py3-none-any.whl (10 kB)
Installing collected packages: zipp, six, pyparsing, packaging, kiwisolver, fonttools, cycler, contourpy, python-dateutil, importlib-resources, matplotlib
Successfully installed contourpy-1.3.0 cycler-0.12.1 fonttools-4.59.2 importlib-resources-6.5.2 kiwisolver-1.4.7 matplotlib-3.9.4 packaging-25.0 pyparsing-3.2.3 python-dateutil-2.9.0.post0 six-1.17.0 zipp-3.23.0(timm_final) E:\AI\Code\OpenSourceCode\G2L-Stereo>pip install tensorboard
Collecting tensorboardUsing cached tensorboard-2.20.0-py3-none-any.whl.metadata (1.8 kB)
Collecting absl-py>=0.4 (from tensorboard)Using cached absl_py-2.3.1-py3-none-any.whl.metadata (3.3 kB)
Collecting grpcio>=1.48.2 (from tensorboard)Downloading grpcio-1.74.0-cp39-cp39-win_amd64.whl.metadata (4.0 kB)
Collecting markdown>=2.6.8 (from tensorboard)Downloading markdown-3.8.2-py3-none-any.whl.metadata (5.1 kB)
Requirement already satisfied: numpy>=1.12.0 in d:\programdata\anaconda3\envs\timm_final\lib\site-packages (from tensorboard) (2.0.2)
Requirement already satisfied: packaging in d:\programdata\anaconda3\envs\timm_final\lib\site-packages (from tensorboard) (25.0)
Requirement already satisfied: pillow in d:\programdata\anaconda3\envs\timm_final\lib\site-packages (from tensorboard) (11.3.0)
Collecting protobuf!=4.24.0,>=3.19.6 (from tensorboard)Downloading protobuf-6.32.0-cp39-cp39-win_amd64.whl.metadata (593 bytes)
Requirement already satisfied: setuptools>=41.0.0 in d:\programdata\anaconda3\envs\timm_final\lib\site-packages (from tensorboard) (78.1.1)
Collecting tensorboard-data-server<0.8.0,>=0.7.0 (from tensorboard)Using cached tensorboard_data_server-0.7.2-py3-none-any.whl.metadata (1.1 kB)
Collecting werkzeug>=1.0.1 (from tensorboard)Using cached werkzeug-3.1.3-py3-none-any.whl.metadata (3.7 kB)
Collecting importlib-metadata>=4.4 (from markdown>=2.6.8->tensorboard)Downloading importlib_metadata-8.7.0-py3-none-any.whl.metadata (4.8 kB)
Requirement already satisfied: zipp>=3.20 in d:\programdata\anaconda3\envs\timm_final\lib\site-packages (from importlib-metadata>=4.4->markdown>=2.6.8->tensorboard) (3.23.0)
Collecting MarkupSafe>=2.1.1 (from werkzeug>=1.0.1->tensorboard)Downloading MarkupSafe-3.0.2-cp39-cp39-win_amd64.whl.metadata (4.1 kB)
Using cached tensorboard-2.20.0-py3-none-any.whl (5.5 MB)
Using cached tensorboard_data_server-0.7.2-py3-none-any.whl (2.4 kB)
Using cached absl_py-2.3.1-py3-none-any.whl (135 kB)
Downloading grpcio-1.74.0-cp39-cp39-win_amd64.whl (4.5 MB)━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 4.5/4.5 MB 6.8 MB/s  0:00:01
Downloading markdown-3.8.2-py3-none-any.whl (106 kB)
Downloading importlib_metadata-8.7.0-py3-none-any.whl (27 kB)
Downloading protobuf-6.32.0-cp39-cp39-win_amd64.whl (435 kB)
Using cached werkzeug-3.1.3-py3-none-any.whl (224 kB)
Downloading MarkupSafe-3.0.2-cp39-cp39-win_amd64.whl (15 kB)
Installing collected packages: tensorboard-data-server, protobuf, MarkupSafe, importlib-metadata, grpcio, absl-py, werkzeug, markdown, tensorboard
Successfully installed MarkupSafe-3.0.2 absl-py-2.3.1 grpcio-1.74.0 importlib-metadata-8.7.0 markdown-3.8.2 protobuf-6.32.0 tensorboard-2.20.0 tensorboard-data-server-0.7.2 werkzeug-3.1.3(timm_final) E:\AI\Code\OpenSourceCode\G2L-Stereo>pip install scikit-image
Collecting scikit-imageDownloading scikit_image-0.24.0-cp39-cp39-win_amd64.whl.metadata (14 kB)
Requirement already satisfied: numpy>=1.23 in d:\programdata\anaconda3\envs\timm_final\lib\site-packages (from scikit-image) (2.0.2)
Collecting scipy>=1.9 (from scikit-image)Downloading scipy-1.13.1-cp39-cp39-win_amd64.whl.metadata (60 kB)
Collecting networkx>=2.8 (from scikit-image)Downloading networkx-3.2.1-py3-none-any.whl.metadata (5.2 kB)
Requirement already satisfied: pillow>=9.1 in d:\programdata\anaconda3\envs\timm_final\lib\site-packages (from scikit-image) (11.3.0)
Collecting imageio>=2.33 (from scikit-image)Using cached imageio-2.37.0-py3-none-any.whl.metadata (5.2 kB)
Collecting tifffile>=2022.8.12 (from scikit-image)Downloading tifffile-2024.8.30-py3-none-any.whl.metadata (31 kB)
Requirement already satisfied: packaging>=21 in d:\programdata\anaconda3\envs\timm_final\lib\site-packages (from scikit-image) (25.0)
Collecting lazy-loader>=0.4 (from scikit-image)Using cached lazy_loader-0.4-py3-none-any.whl.metadata (7.6 kB)
Downloading scikit_image-0.24.0-cp39-cp39-win_amd64.whl (12.9 MB)━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 12.9/12.9 MB 9.2 MB/s  0:00:01
Using cached imageio-2.37.0-py3-none-any.whl (315 kB)
Using cached lazy_loader-0.4-py3-none-any.whl (12 kB)
Downloading networkx-3.2.1-py3-none-any.whl (1.6 MB)━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 1.6/1.6 MB 9.9 MB/s  0:00:00
Downloading scipy-1.13.1-cp39-cp39-win_amd64.whl (46.2 MB)━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 46.2/46.2 MB 11.0 MB/s  0:00:04
Downloading tifffile-2024.8.30-py3-none-any.whl (227 kB)
Installing collected packages: tifffile, scipy, networkx, lazy-loader, imageio, scikit-image
Successfully installed imageio-2.37.0 lazy-loader-0.4 networkx-3.2.1 scikit-image-0.24.0 scipy-1.13.1 tifffile-2024.8.30(timm_final) E:\AI\Code\OpenSourceCode\G2L-Stereo>pip install opencv-python
Collecting opencv-pythonUsing cached opencv_python-4.12.0.88-cp37-abi3-win_amd64.whl.metadata (19 kB)
Requirement already satisfied: numpy<2.3.0,>=2 in d:\programdata\anaconda3\envs\timm_final\lib\site-packages (from opencv-python) (2.0.2)
Using cached opencv_python-4.12.0.88-cp37-abi3-win_amd64.whl (39.0 MB)
Installing collected packages: opencv-python
Successfully installed opencv-python-4.12.0.88

文章转载自：

http://00000000.wnjrf.cn
http://00000000.wnjrf.cn
http://00000000.wnjrf.cn
http://00000000.wnjrf.cn
http://00000000.wnjrf.cn
http://00000000.wnjrf.cn
http://00000000.wnjrf.cn
http://00000000.wnjrf.cn
http://00000000.wnjrf.cn
http://00000000.wnjrf.cn
http://00000000.wnjrf.cn
http://00000000.wnjrf.cn
http://00000000.wnjrf.cn
http://00000000.wnjrf.cn
http://00000000.wnjrf.cn
http://00000000.wnjrf.cn
http://00000000.wnjrf.cn
http://00000000.wnjrf.cn
http://00000000.wnjrf.cn
http://00000000.wnjrf.cn
http://00000000.wnjrf.cn
http://00000000.wnjrf.cn
http://00000000.wnjrf.cn
http://00000000.wnjrf.cn
http://00000000.wnjrf.cn
http://00000000.wnjrf.cn
http://00000000.wnjrf.cn
http://00000000.wnjrf.cn
http://00000000.wnjrf.cn
http://00000000.wnjrf.cn
http://www.dtcms.com/a/367782.html

相关文章：

  • UIViewController生命周期
  • 大文件断点续传解决方案：基于Vue 2与Spring Boot的完整实现
  • 商城系统——项目测试
  • Ubuntu镜像源配置
  • 【C语言】第二课 基础语法
  • 机器学习基础-day07-项目案例
  • 无开机广告，追觅一口气推出三大系列高端影音新品该咋看？
  • Vben5 自带封装好的组件（豆包版）
  • 漏洞修复 Nginx SSL/TLS 弱密码套件
  • IDEA终极配置指南：打造你的极速开发利器
  • maven settings.xml文件的各个模块、含义以及它们之间的联系
  • 一文详解大模型强化学习（RLHF）算法：PPO、DPO、GRPO、ORPO、KTO、GSPO
  • websocket的key和accept分别是多少个字节
  • lc链表问答
  • [iOS] 折叠 cell
  • Qt 系统相关 - 1
  • JavaScript 实战进阶续篇：从工程化到落地的深度实践
  • 深度学习：自定义数据集处理、数据增强与最优模型管理
  • ASRPRO语音模块
  • 一个开源的企业官网简介
  • Linux的权限详解
  • 【ICCV 2025 顶会论文】，新突破！卷积化自注意力 ConvAttn 模块，即插即用，显著降低计算量和内存开销。
  • HTB Jerry
  • 微信支付--在线支付实战，引入Swagger,定义统一结果，创建并连接数据库
  • 为什么串口发送一串数据时需要延时？
  • 决策树算法详解：从原理到实战
  • 生成式AI优化新纪元：国产首个GEO工具的技术架构剖析
  • 2025年高教社杯全国大学生数学建模竞赛B题思路（2025数学建模国赛B题思路）
  • 【C语言】第一课 环境配置
  • git命令行打patch