01-1 搭建 pytorch 虚拟环境

pytorch 管网:PyTorch


一 进入 Anaconda

二 创建虚拟环境

python 复制代码
conda create -n pytorch python=3.9
  1. 注意要注意断 VPN
  2. 切换镜像:
  • 移除原来的镜像
python 复制代码
# 查看当前配置
conda config --show channels
conda config --show-sources

# 移除之前的镜像
conda config --remove-key channels.https://pypi.tuna.tsinghua.edu.cn/simple
  • 添加新的镜像
python 复制代码
conda config --add channels https://mirrors.aliyun.com/anaconda/pkgs/main/
conda config --add channels https://mirrors.aliyun.com/anaconda/pkgs/free/
conda config --add channels https://mirrors.aliyun.com/anaconda/cloud/pytorch/
conda config --add channels https://mirrors.aliyun.com/anaconda/cloud/conda-forge/

三 进入虚拟环境

python 复制代码
conda activate pytorch

3.1 直接使用官网的方式

Anaconda也可以使用 pip 的方式安装,大小我这个 2G左右

多测试下,网络问题试了几次。

复制代码
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu117

后面是错误尝试:


四 在虚拟环境中安装 pytorch

python 复制代码
conda install pytorch torchvision torchaudio cudatoolkit=11.0 -c pytorch -c nvidia

conda install pytorch==1.7.0 torchvision==0.8.1 torchaudio -c pytorch

# 11.6
conda install pytorch==1.12.0 torchvision==0.13.0 torchaudio==0.12.0 cudatoolkit=11.6 -c pytorch -c conda-forge

conda install pytorch==1.9.0 torchvision==0.10.0 torchaudio==0.9.0 cudatoolkit=11.3 -c pytorch -c conda-forge

# CUDA 11.0
conda install pytorch==1.7.0 torchvision==0.8.0 torchaudio==0.7.0 cudatoolkit=11.0 -c pytorch 

conda install pytorch torchvision torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia

pytorch:1.9.0 torchvision: 0.10.0 python: 3.9

下载地址:Index of /anaconda/cloud/pytorch/win-64/ | 清华大学开源软件镜像站

查看cuda版本

python 复制代码
nvidia-smi

查看cuda 版本

python 复制代码
nvcc --version

nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2020 NVIDIA Corporation
Built on Thu_Jun_11_22:26:48_Pacific_Daylight_Time_2020
Cuda compilation tools, release 11.0, V11.0.194
Build cuda_11.0_bu.relgpu_drvr445TC445_37.28540450_0

RTX 3080的算力是8.6,3080Ti只能适用CUDA11.0以上的版本;需要重新安装对应CUDA可用版本的pytorch版本;

pip之后默认安装的环境是torch1.9.0+cuda10.2,而3080ti的算力是8.6,支持的cuda版本应该不小于11.0,所以出错。

五 使用安装包安装

python 复制代码
conda activate pytorch

# 跳转文件夹
cd /d E:\4-viev_find\yolo5
python 复制代码
conda install --offline pytorch-1.9.0-py3.9_cuda11.1_cudnn8_0.tar.bz2

conda install --use-local pytorch-1.9.0-py3.9_cuda11.1_cudnn8_0.tar.bz2 -n pytorch
相关推荐
青松@FasterAI13 分钟前
【程序员 NLP 入门】词嵌入 - 上下文中的窗口大小是什么意思? (★小白必会版★)
人工智能·自然语言处理
AIGC大时代28 分钟前
高效使用DeepSeek对“情境+ 对象 +问题“型课题进行开题!
数据库·人工智能·算法·aigc·智能写作·deepseek
硅谷秋水29 分钟前
GAIA-2:用于自动驾驶的可控多视图生成世界模型
人工智能·机器学习·自动驾驶
偶尔微微一笑41 分钟前
AI网络渗透kali应用(gptshell)
linux·人工智能·python·自然语言处理·编辑器
深度之眼1 小时前
2025时间序列都有哪些创新点可做——总结篇
人工智能·深度学习·机器学习·时间序列
晓数1 小时前
【硬核干货】JetBrains AI Assistant 干货笔记
人工智能·笔记·jetbrains·ai assistant
jndingxin1 小时前
OpenCV 图形API(60)颜色空间转换-----将图像从 YUV 色彩空间转换为 RGB 色彩空间函数YUV2RGB()
人工智能·opencv·计算机视觉
Sherlock Ma2 小时前
PDFMathTranslate:基于LLM的PDF文档翻译及双语对照的工具【使用教程】
人工智能·pytorch·语言模型·pdf·大模型·机器翻译·deepseek
知舟不叙2 小时前
OpenCV中的SIFT特征提取
人工智能·opencv·计算机视觉
船长@Quant2 小时前
文档构建:Sphinx全面使用指南 — 基础篇
python·markdown·sphinx·文档构建