linux环境下pytorch（GPU版本）的安装

print(torch.cuda.is_available())# 能否调用cuda Ture/False。print(torch.__version__)# torch版本。anaconda -V（注意a要小写，V要大写）；conda -V ,显示版本信息。1.安装anaconda。3.安装pytorch。

qq_52259359

1620人浏览 · 2024-03-12 11:25:47

qq_52259359 · 2024-03-12 11:25:47 发布

1.安装anaconda

https://repo.anaconda.com/archive/
bash Anaconda3-2023.09-0-Linux-x86_64.sh

2.配置环境

xbgydx@ubuntu:~$ nano ~/.bashrc
xbgydx@ubuntu:~$ export PATH="/home/xbgydx/anaconda3/bin:$PATH"
按 Ctrl + X，然后按 Y 保存更改，最后按 Enter 退出。

source ~/.bashrc

anaconda -V（注意a要小写，V要大写）；
conda -V ,显示版本信息

conda init bash
conda activate pytorch

3.安装pytorch

conda create -n pytorch python=3.9
conda info --envs

nvcc -V
sudo apt install nvidia-cuda-toolkit

可选：

~/.bashrc

export PATH=/usr/local/cuda/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH
source ~/.bashrc

pip install torch==1.8.1+cu101 torchvision==0.9.1+cu101 torchaudio==0.8.1 -f https://download.pytorch.org/whl/torch_stable.html  -i  https://pypi.tuna.tsinghua.edu.cn/simple

4.检测是否成功

python
import torch
print(torch.__version__) # torch版本
print(torch.cuda.is_available()) # 能否调用cuda Ture/False

魔乐社区

魔乐社区（Modelers.cn) 是一个中立、公益的人工智能社区，提供人工智能工具、模型、数据的托管、展示与应用协同服务，为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作，由全产业链共同建设、共同运营、共同享有，推动国产AI生态繁荣发展。

更多推荐

【计算机视觉】Pixel逐像素分类&Mask掩码分类理解摘要

魔乐社区

计算机视觉（opencv）实战三十二——CascadeClassifier 人脸微笑检测（摄像头）

本文从原理到实现，详细介绍了基于 OpenCV Haar 分类器的人脸与微笑检测：讲解了 Haar 特征和级联检测原理。对代码逐行拆解并解释参数含义。画出完整流程图，帮助理解执行过程。给出了常见问题和优化建议，甚至扩展到深度学习方法。这种方法简单、轻量、实时性好，非常适合入门和小型应用项目。但如果需要更高准确率和更强鲁棒性，建议使用深度学习检测器替代 Haar 分类器。