首页 > 系统相关 >环境配置·Ubuntu1804安装CUDA和Pytorch

环境配置·Ubuntu1804安装CUDA和Pytorch

时间:2024-06-06 19:33:11浏览次数:21  
标签:Ubuntu1804 whl wsl Pytorch CUDA ubuntu wget local cuda

Init Ubuntu and change deb&pip source

wget https://github.com/blueflylabor/blueflylabor.github.io/blob/main/toolbox/initUbuntu/initUbuntu.sh
bash ./initUbuntu.sh

CUDA11.6

wget https://developer.download.nvidia.com/compute/cuda/repos/wsl-ubuntu/x86_64/cuda-wsl-ubuntu.pin
sudo mv cuda-wsl-ubuntu.pin /etc/apt/preferences.d/cuda-repository-pin-600
wget https://developer.download.nvidia.com/compute/cuda/11.6.2/local_installers/cuda-repo-wsl-ubuntu-11-6-local_11.6.2-1_amd64.deb
sudo dpkg -i cuda-repo-wsl-ubuntu-11-6-local_11.6.2-1_amd64.deb
sudo apt-key add /var/cuda-repo-wsl-ubuntu-11-6-local/7fa2af80.pub
sudo apt-get update
sudo apt-get -y install cuda

Load library path

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

Test nvidia-smi

nvidia-smi

Test nvcc -V

nvcc -V

Pytorch

pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113

get file when network worse

wget https://download.pytorch.org/whl/cu113/torch-1.10.2%2Bcu113-cp36-cp36m-linux_x86_64.whl
wget https://download.pytorch.org/whl/cu113/torchvision-0.11.3%2Bcu113-cp36-cp36m-linux_x86_64.whl
wget 
pip3 install ./torch-1.10.2+cu113-cp36-cp36m-linux_x86_64.whl
pip3 install ./torchvision-0.11.3+cu113-cp36-cp36m-linux_x86_64.whl

标签:Ubuntu1804,whl,wsl,Pytorch,CUDA,ubuntu,wget,local,cuda
From: https://www.cnblogs.com/blueflylabor/p/18235898

相关文章

  • 深度学习 - PyTorch简介
    基础知识1.PyTorch简介PyTorch的特点和优势:动态计算图、易用性、强大的社区支持、与NumPy兼容。安装和环境配置:安装和验证PyTorch:pipinstalltorchtorchvision验证安装:importtorchprint(torch.__version__)运行结果:1.9.0#具体版本可能不同配置虚拟......
  • 麻醉医生的深度学习之旅 P4:Pytorch实现猴痘病识别
    ......
  • 配置Tensorflow使用GPU加速(Tensorflow2.6.0,Python3.6-3.9,CUDA11.2,cuDNN8.1)
    目录前言一、安装Anaconda1.1从Anaconda官网下载Anaconda(默认安装最新版本)1.2安装Anaconda并配置环境变量二、安装CUDA和cuDNN2.1下载CUDAToolkit2.2安装CUDAToolkit2.3下载安装cuDNN2.4配置CUDA环境变量三、创建Anaconda虚拟环境并安装Tensorflow3.1创建......
  • 基于 PyTorch 的 Python 深度学习:注意力机制
    基于PyTorch的Python深度学习:注意力机制深度学习在近年来取得了巨大的进步,而注意力机制(AttentionMechanism)作为其中的一个重要概念,为模型提供了一种捕捉输入数据中不同部分之间关系的能力。在本文中,我们将探讨注意力机制的基本概念,以及如何在PyTorch框架下实现注意......
  • 有Cuda能力的GPU内核
    当CUDA应用程序启动一个内核时,CUDA运行时会确定系统中每个GPU的计算能力,并利用这些信息自动寻找最适合该GPU的内核cubin文件或PTX版本(如果可用)。如果存在支持目标GPU架构的cubin文件,将直接使用它;否则,CUDA运行时将加载PTX代码,并在启动之前将其即时编译(JIT编译)为GPU的本机cubin格式......
  • cuda-toolkit 设置环境变量 CUDA_FORCE_ PTX_JIT的作用
    设置环境变量CUDA_FORCE_PTX_JIT的作用是控制NVIDIACUDA运行时是否强制使用PTX即时编译(Just-In-Timecompilation)。当这个环境变量被设置时,它会对CUDA程序的行为产生以下影响:禁用二进制缓存加载:正常情况下,CUDA运行时会尝试从二进制缓存中加载预先编译好的cubin(CUDA二进制)文......
  • Cuda计算能力
    NVIDIACUDAC++编译器nvcc可用于生成针对特定架构的cubin文件和每个内核的向前兼容的PTX版本。每个cubin文件针对特定的计算能力版本,并且仅与具有相同主版本号的GPU架构向前兼容。例如,针对计算能力3.0的cubin文件在所有计算能力3.x(Kepler)设备上受支持,但在计算能力5.x(Maxwell)设备......
  • Pytorch根据论文手搓CoPE
    根据论文《ContextualPositionEncoding: LearningtoCountWhat’sImportant》编写的CoPE代码。具备多头计算能力,本人水平不高,个人代码未经验证,有问题和建议欢迎指出。CoPE代码为代替LlamaModel中的RoPE而设计,因此函数名称类似于LlamaModel的RoPE,但实际功能略有不同,有待商......
  • 解决安装cuda时vs报错MSB3721的问题
    以下几种可能1.调整调试平台为x642.点击vs顶部导航栏的调试->最下面那一项属性->CUDAC/C++->Common->CUDADustuomDir设置为你安装CUDAtoolkit的目录3.如果你的电脑系统用户名为中文,请前往环境变量->用户变量,将tmp和temp的值修改为一个新文件夹(建议在D盘里面创建一个tem......
  • Windows10系统中安装与配置PyTorch(无GPU版本)
    文章目录1.什么是PyTorch2.PyTorch的安装与配置(无GPU)2.1创建环境2.2安装pytorch库(无GPU)2.3验证安装结果1.什么是PyTorchPyTorch是一种用于构建深度学习模型且功能完备的开源框架,通常用于处理图像识别和语言处理等应用当中的机器学习。PyTorch保留了Torch(L......