conda pytorch 安装使用

时间:2024-11-12 11:36:20

环境备注

win11 专业版

pycharm

anaconda 3.1

python: python3.8

pytorch: 显卡是RTX 4060 所以对应选择版本:torch2.4.1+cu118

torchvision: 根据 pytorch选择对应版本(torchvision 0.19.1+cu118 注意这里直接安装会安装不带cu118的,也就是CPU的),一般使用conda默认安装即可,额根据参考检擦版本是否对应

2.4版本pytorch对应0.19版本torchvision 3.8版本python

参考:

https://github.com/pytorch/vision 做好版本对应!

torchvision 版本对应下载地址: download.pytorch.org/whl/torch_stable.html

需要安装的库:

pip install numpy
pip install scipy
pip install matplotlib
pip install opencv-python
pip install tqdm
pip install wheel
pip install lxml
pip install pillow==9.5.0   ##版本再高点的话imgsize绘图函数就用不了了
# 这里需要特别安装带有cu118版本的库,直接安装会默认安装不带cu118的
pip install torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

1. conda 使用笔记

conda常用使用方法记录

1.1 基础

创建环境

查看当前conda中所有的虚拟环境

conda env list

创建一个 Python 3.7 版本的 Conda 虚拟环境,可以使用以下命令:

conda create --name env-pytorch python=3.6

要进入(激活)一个目标 Conda 虚拟环境,可以使用以下命令:

conda activate env-pytorch

要退出当前环境,可以使用以下命令:

conda deactivate

假设你想要将名为 myenv 的环境重命名为 mynewenv

conda create --name mynewenv --clone myenv
conda remove --name myenv --all

2. 安装pytorch

Pytorch 、torchvision、python版本对应图

在这里插入图片描述

首先恢复和清除所有之前的源

conda config --remove-key channels

查看源

conda config --show

创建Anaconda配置文件.condarc

conda config --set show_channel_urls yes  # 创建Anaconda配置文件.condarc

更换源

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
conda config --set show_channel_urls yes

来到虚拟环境安装pytorch

从本地开始 |PyTorch 插件

conda create --name py3.8-touch python=3.8
conda activate py3.8-touch
pip install torch==1.13.1+cu117 torchvision==0.14.1+cu117 torchaudio==0.13.1 -f https://download.pytorch.org/whl/torch_stable.html -i https://pypi.tuna.tsinghua.edu.cn/simple
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
pip install torch==1.13.1 torchvision==0.14.1 torchaudio==0.13.1

pip install 安装 torch cuda 11.8 cu118_pip install torch==2.3.1 torchvision==0.18.1 torch-****博客

解决报错:

CondaValueError: Malformed version string ‘~’: invalid character(s).

解决问题Conda:CondaValueError: Malformed version string ‘~’ : invalid character(s)_condavalueerror: malformed version string ‘~’: inv-****博客

E:\app\Anaconda3\envs\pytorch
E:\app\Anaconda3\envs\env-python3.7-pytorch

测试pytorch

python
import torch
x = torch.rand(5, 3)
print(x)

此外,要检查您的 GPU 驱动程序和 CUDA 是否已启用并可通过 PyTorch 访问,请运行以下命令以返回 CUDA 驱动程序是否已启用:

import torch
torch.cuda.is_available()