系統(tǒng)環(huán)境中:
- 1.系統(tǒng)驅(qū)動安裝的是:
NVIDIA-Linux-x86_64-530.30.02.run文章來源地址http://www.zghlxwxcb.cn/news/detail-685987.html
- 2.CUDA安裝:cuda_12.1.0_530.30.02_linux.run(無需第1步,直接安裝它就帶配套驅(qū)動)
wget https://developer.download.nvidia.com/compute/cuda/12.1.0/local_installers/cuda_12.1.0_530.30.02_linux.run
sudo sh cuda_12.1.0_530.30.02_linux.run
- 3.CUDNN安裝:cudnn-linux-x86_64-8.8.1.3_cuda12-archive.tar.xz
sudo cp ./include/* /usr/local/cuda-12.1/include
sudo cp ./lib/libcudnn* /usr/local/cuda-12.1/lib64
sudo chmod a+r /usr/local/cuda-12.1/include/cudnn*?
sudo chmod a+r /usr/local/cuda-12.1/lib64/libcudnn*
Anaconda虛擬環(huán)境中
# 先創(chuàng)建虛擬環(huán)境
conda create --name Dogs python=3.9
# 安裝cudatoolkit-11.8.0+cudnn-8.9.2.26
conda install -c main cudatoolkit=11.8 cudnn=8.9.2.26
# pytorch安裝
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
文章來源:http://www.zghlxwxcb.cn/news/detail-685987.html
到了這里,關(guān)于Ubuntu22.04 LTS+NVIDIA 4090+Cuda12.1+cudnn8.8.1的文章就介紹完了。如果您還想了解更多內(nèi)容,請在右上角搜索TOY模板網(wǎng)以前的文章或繼續(xù)瀏覽下面的相關(guān)文章,希望大家以后多多支持TOY模板網(wǎng)!