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