pytorch安裝-國内鏡像源

大家好，又見面了，我是你們的朋友全棧君。

在安裝好cuda和cudnn之後安裝pytorch的方法

cudnn安裝：

1. 到cudnn官網下載下傳：cudnn官方

   下載下傳cuDNN Library for Linux (x86_64)

2. 解壓縮：

cp  cudnn-8.0-linux-x64-v5.1.solitairetheme8 cudnn-8.0-linux-x64-v5.1.tgz
tar -xvf cudnn-8.0-linux-x64-v5.1.tgz           

複制

1. 拷貝檔案

sudo cp cuda/include/cudnn.h /usr/local/cuda/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*
sudo ldconfig           

複制

1. 檢驗

   cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

安裝pytorch：

網上很多的方法都不是鏡像下載下傳，或者鏡像下載下傳因為系統的問題找不到庫

打開官網，找到對應合适的版本（cuda)：

https://pytorch.org/get-started/locally/

之後複制下面這一行指令：

conda install pytorch torchvision torchaudio cudatoolkit=11.0 -c pytorch

接下來就是關鍵一步了，

把-c pytorch表示的pytorch源，更改為國内的鏡像。

https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/

先浏覽器打開這個網頁，然後選擇你的系統

點開之後複制這個新位址，将-c pytorch更改為

conda install pytorch torchvision torchaudio cudatoolkit=11.0 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/linux-64/           

複制

完畢

檢驗步驟：

import torch
print(torch.__version__)

print(torch.version.cuda)
print(torch.backends.cudnn.version())

torch.cuda.is_available()
#cuda是否可用；

torch.cuda.device_count()
#傳回gpu數量；

torch.cuda.get_device_name(0)
#傳回gpu名字，裝置索引預設從0開始；

torch.cuda.current_device()
#傳回目前裝置索引           

複制

有問題歡迎留言交流~

cuda安裝教程：https://www.csdn.net/tags/Mtjacg5sOTA5NTctYmxvZwO0O0OO0O0O.html

釋出者：全棧程式員棧長，轉載請注明出處：https://javaforall.cn/152757.html原文連結：https://javaforall.cn