conda pytorch 安装使用
环境备注
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
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-CSDN博客
解决报错:
CondaValueError: Malformed version string ‘~’: invalid character(s).
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()
原文地址:https://blog.csdn.net/qq_44179528/article/details/143676324
免责声明:本站文章内容转载自网络资源,如本站内容侵犯了原著者的合法权益,可联系本站删除。更多内容请关注自学内容网(zxcms.com)!