ubuntu18+cuda10.1+mindspore1.2.1

转载地址：https://bbs.huaweicloud.com/forum/thread-133322-1-1.html

作者：升级失败

1.ubuntu18 安装Nvidia显卡驱动

sudo apt remove --purge nvidia*                      # 卸载已有的nvidia显卡驱动（如果已安装的话）
sudo add-apt-repository ppa:graphics-drivers/ppa     # 添加ppa源
sudo apt update                                      # 更新源列表
ubuntu-drivers devices                               # 查看可安装的驱动列表（见下图选择recommended那项进行安装）
sudo apt install nvidia-driver-460                   # 安装nvidia显卡驱动
sudo reboot                                          # 重新启动电脑
# 推荐三种方式检查自己nvidia显卡驱动是否安装成功
nvidia-smi                                           # 检查nvidia显卡的相关信息
nvidia-settings                                      # 查看nvidia设置面板

2. python3.7.5 cuda10.1 cudnn7.6.5 gmp6.1.2

GMP高精度数学库
wget ftp://ftp.gnu.org/gnu/gmp/gmp-6.1.2.tar.xz
xz -d gmp-6.1.2.tar.xz
tar -xvf gmp-6.1.2.tar
cd gmp-6.1.2
./configure --enable-cxx
make -j8
make install

安装CUDA 10.1 cudnn

sudo chmod +x cuda_10.1.105_418.39_linux.run     # 给安装文件赋予执行权限
sudo ./cuda_10.1.105_418.39_linux.run

系统会提示要修改PATH和LD_LIBRARY_PATH.

tar -zxvf cudnn-10.1-linux-x64-v7.6.5.32.tgz
cd cuda
cp ./include/* /usr/local/cuda-10.1/include
cp ./lib64/*   /usr/local/cuda-10.1/lib64

安装mindspore的gpu版本

pip3.7 install https://ms-release.obs.cn-north-4.myhuaweicloud.com/1.2.1/MindSpore/gpu/ubuntu_x86/cuda-10.1/mindspore_gpu-1.2.1-cp37-cp37m-linux_x86_64.whl --trusted-host ms-release.obs.cn-north-4.myhuaweicloud.com -i https://pypi.tuna.tsinghua.edu.cn/simple

验证是否成功安装

vim test.py

import numpy as np
from mindspore import Tensor
import mindspore.ops as ops
import mindspore.context as context

context.set_context(device_target="GPU")
x = Tensor(np.ones([1,3,3,4]).astype(np.float32))
y = Tensor(np.ones([1,3,3,4]).astype(np.float32))
print(ops.add(x, y))

Ubuntu 18.04 安装 PyCharm

下载链接 https://www.jetbrains.com/pycharm/download/#section=linux

tar -zxvf pycharm-community-2021.1.2.tar.gz
cd pycharm-community-2021.1.2/bin
./pycharm.sh

创建新工程：指定系统中安装的python3.7.5

引用[https://bbs.huaweicloud.com/blogs/198357]