网页做得好的网站,做网站的所有代码,石家庄百度推广排名,泸州市建设厅网站一、显卡驱动安装 执行nvidia-smi查看安装情况 二、cuda安装 cuda官网下载cuda_11.6.2_510.47.03_linux.run#xff0c;安装执行
sudo sh cuda_11.6.2_510.47.03_linux.run提升安装项#xff0c;驱动不用安装#xff0c;即第一项#xff08;Driver#xff09;#xff…一、显卡驱动安装 执行nvidia-smi查看安装情况 二、cuda安装 cuda官网下载cuda_11.6.2_510.47.03_linux.run安装执行
sudo sh cuda_11.6.2_510.47.03_linux.run提升安装项驱动不用安装即第一项Driver按空格键取消接下选在Install进行安装 修改环境变量
sudo gedit ~/.bashrc在最后添加
# Add the related bin and lib for cuda toolkit
export PATH/usr/local/cuda-11.6/bin:$PATH
export LD_LIBRARY_PATH/usr/local/cuda-11.6/lib64:$LD_LIBRARY_PATH更新环境
source ~/.bashrc查看安装情况
nvcc -V三、安装cudnn cudnn官网下载cudnn-local-repo-ubuntu1804-8.9.5.30_1.0-1_amd64.deb官网安装教程
sudo dpkg -i cudnn-local-repo-ubuntu1804-8.9.5.30_1.0-1_amd64.debcd /var/cudnn-local-repo-ubuntu1804-8.9.5.30
sudo apt-get install /var/cudnn-local-repo-ubuntu1804-8.9.5.30/libcudnn8_8.9.5.30-1cuda11.8_amd64.deb
sudo apt-get install /var/cudnn-local-repo-ubuntu1804-8.9.5.30/libcudnn8-dev_8.9.5.30-1cuda11.8_amd64.deb
sudo apt-get install /var/cudnn-local-repo-ubuntu1804-8.9.5.30/libcudnn8-samples_8.9.5.30-1cuda11.8_amd64.deb四、安装tensorrt 官网下载TensorRT-8.5.1.7.Linux.x86_64-gnu.cuda-11.8.cudnn8.6.tar.gz 解压文件拷贝到对应文件下
sudo mv TensorRT-8.5.1.7/ /opt/
sudo gedit ~/.bashrcbashrc文件最下方添加
export PATH/opt/TensorRT-8.5.1.7/bin:$PATH
export LD_LIBRARY_PATH/opt/TensorRT-8.5.1.7/lib:$LD_LIBRARY_PATH激活环境
source .bashrc测试
cd /opt/TensorRT-8.5.1.7/samples/sampleOnnxMNIST
make -j2
cd /opt/TensorRT-8.5.1.7/bin
./sample_onnx_mnist如果出现预测结果说明安装成功
五、anconda安装 下载Anaconda3-2021.04-Linux-x86_64.sh进行安装
sudo sh Anaconda3-2021.04-Linux-x86_64.sh如果想关闭默认环境执行
If youd prefer that condas base environment not be activated on startup, set the auto_activate_base parameter to false: conda config --set auto_activate_base false
sudo chmod aw /home/ubuntu/.conda六、pytorch-gpu安装 安装链接执行
pip install torch1.13.0cu116 torchvision0.14.0cu116 torchaudio0.13.0 --extra-index-url https://download.pytorch.org/whl/cu116七、pycharm安装 下载并安装 添加快捷方式
sudo gedit /usr/share/applications/pycharm.desktop[Desktop Entry]
Version1.0
NamePycharm Community Edition
CommentPython IDE for Professional Developers
Exec/xxx/pycharm/bin/pycharm.sh
Icon/xxx/pycharm/bin/pycharm.png
Terminalfalse
StartupNotifytrue
StartupWMClassjetbrains-pycharm-ce
TypeApplication
CategoriesDevelopment;IDE;Python;执行
sh /xxx/pycharm/bin/pycharm.sh运行pycharm应用
环境添加选在python3.xx