1.更新驅動
(1)查看驅動版本
1 ubuntu-drivers devices

(2)安裝對應的驅動

1 sudo apt install nvidia-430 已經安裝過了,若未安裝,會進行安裝.

參考:https://zhuanlan.zhihu.com/p/59618999
2.安裝Anaconda
https://www.anaconda.com/products/individual

(1) 1 sudo bash Anaconda3-2020.02-Linux-x86_64.sh 文件名根據自己下載的版本確定
(2)安裝完成后配置環境變量
1 sudo gedit ~/.bashrc

把echo export PATH="~/anaconda3/bin:$PATH">> ~/.bashrc添加到最后一行

(3)終端輸入python3出現下圖即安裝成功

(3)創建一個名為pytorch的虛擬環境
conda create -n pytorch python=3.7
注意:在創建環境的時候遇到了如下錯誤:

寫入如下代碼即可解決
1 sudo chmod 777 -R ~/anaconda3/
2 sudo chmod 777 -R ~/.conda/
激活虛擬環境
1 source activate pytorch

3.安裝pytorch
(1)ubuntu已更換過鏡像源,在安裝過程中仍然很慢。
配置源:
sudo gedit ~/.condarc
配置文件修改為:
1 channels: 2 - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/ 3 - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/menpo/ 4 - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/bioconda/ 5 - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/ 6 - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/ 7 - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/ 8 - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/ 9 - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge 10 show_channel_urls: true
(2)命令行安裝pytorch(注意一定去掉-c):

1 conda install pytorch torchvision cudatoolkit=10.1
參考:1.配置源參考https://blog.csdn.net/watermelon1123/article/details/88122020
2.若未更換鏡像源可參考:https://blog.csdn.net/weixin_43876206/article/details/100924378
(3)測試pytorch是否安裝成功
1 source activate pytorch
pytorch是我建立的虛擬環境名,可根據自己的進行修改
2 python3
3 import torch
4 torch.cuda.is_available()

4.安裝pycharm
具體安裝可參考https://blog.csdn.net/qq_38451119/article/details/81023766
(1)下載Pycharm安裝包,提取出來
提取完畢后在pycharm的bin文件夾下打開終端,輸入
1 ./pycharm.sh 進行安裝即可.
(2)安裝完畢,進行環境配置,
one:.打開pycharm, 依次點擊 file - > Settings- >Project: xx - >Project Interpreter
two:點擊右上角的小齒輪,點擊彈出來的Add Python Interpreter:
three:選擇anconda,選中穿件的anconda (pytorch)環境,保存即可

(3)測試:

