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
相关推荐
凯子坚持 c几秒前
基于飞桨框架3.0本地DeepSeek-R1蒸馏版部署实战
人工智能·paddlepaddle
老歌老听老掉牙5 分钟前
平面旋转与交线投影夹角计算
python·线性代数·平面·sympy
满怀101510 分钟前
Python入门(7):模块
python
无名之逆10 分钟前
Rust 开发提效神器:lombok-macros 宏库
服务器·开发语言·前端·数据库·后端·python·rust
你觉得20514 分钟前
哈尔滨工业大学DeepSeek公开课:探索大模型原理、技术与应用从GPT到DeepSeek|附视频与讲义下载方法
大数据·人工智能·python·gpt·学习·机器学习·aigc
啊喜拔牙22 分钟前
1. hadoop 集群的常用命令
java·大数据·开发语言·python·scala
8K超高清31 分钟前
中国8K摄像机:科技赋能文化传承新图景
大数据·人工智能·科技·物联网·智能硬件
hyshhhh1 小时前
【算法岗面试题】深度学习中如何防止过拟合?
网络·人工智能·深度学习·神经网络·算法·计算机视觉
薛定谔的猫-菜鸟程序员1 小时前
零基础玩转深度神经网络大模型:从Hello World到AI炼金术-详解版(含:Conda 全面使用指南)
人工智能·神经网络·dnn
币之互联万物1 小时前
2025 AI智能数字农业研讨会在苏州启幕,科技助农与数据兴业成焦点
人工智能·科技