Ubuntu20.04+RTX3090+Nvidia460+cuda11.1+Anaconda3+pytorch1.8.1


1、查看是否安裝nvidia驅動:sudo lshw -c video 

(configuration: driver=nvidia latency=0)

2、查看NVIDIA內核:cat /proc/driver/nvidia/version

查看NVIDIA驅動版本:sudo dpkg --list | grep nvidia-*

3、安裝NVDIA驅動:sudo apt install nvidia-driver-465

    必須重啟:reboot

4、卸載NVDIA驅動:

sudo apt-get --purge remove nvidia*

sudo apt-get --purge remove "*cublas*" "cuda*"

sudo apt-get --purge remove "*nvidia*"

sudo rm -rf /usr/local/cuda-xx.x

 

5、cuda下載:

(root環境下裝好各個包,其他用戶增加配置即可vim ~/.bashrc)

https://developer.nvidia.com/cuda-toolkit-archive

修改配置:

sudo vim ~/.bashrc

最后添加:

export CUDA_HOME=/usr/local/cuda

export PATH=$PATH:$CUDA_HOME/bin

export LD_LIBRARY_PATH=/usr/local/cuda-10.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

保存:

source ~/.bashrc

測試:

cd /usr/local/cuda/samples/1_Utilities/deviceQuery

sudo make

./deviceQuery

查看版本:

cat  /usr/local/cuda/version.txt或nvcc -V

卸載:

sudo /usr/local/cuda-10.2/bin/cuda-uninstaller

rm -rf /usr/local/cuda-10.2/

6、cudnn下載:

(安裝多個環境:環境配置里切換cuda,cudnn則是直接重裝,復制文件夾)

https://developer.nvidia.com/rdp/cudnn-archive#a-collapse820-111

tar zxvf cudnn-10.0-linux-x64-v7.6.5.32.tgz -C ./

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 

sudo chmod a+r /usr/local/cuda/lib64/libcudnn*

測試:

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

 

7、安裝anaconda

https://www.anaconda.com/products/individual-b

bash Anaconda2-2019.07-Linux-x86_64.sh

重開終端

conda env list

8、卸載anaconda

rm -rf anaconda3   #文件夾名

(vi ~/.bashrc 刪除相關語句)

9、創建python37環境

conda create -n py37 python=3.7

source activate py37

刪除環境:conda remove -n py37 --all

 

10、安裝pytorch

https://pytorch.org/get-started/previous-versions/

pip3 install torch==1.8.1+cu111 torchvision==0.9.1+cu111 torchaudio==0.8.1 -f https://download.pytorch.org/whl/torch_stable.html

使用pip安裝 pip install torch==1.8.0


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM