环境准备
Python版本控制-3.7.5
此处采用pyenv来创建MindSpore环境,首先查看是否有3.7.5版本:
pyenv versions
CUDA-10.1
尝试过直接使用CUDA10.2,但是在运行from mindspore import Tensor时会出现ImportError: libcudart.so.10.1: cannot open shared object file: No such file or directory。所以还是要安装CUDA10.1。
首先进入官网根据系统版本进行选择:https://developer.nvidia.com/cuda-10.1-download-archive-update2?target_os=Linux&target_arch=x86_64&target_distro=Ubuntu&target_version=1804&target_type=runfilelocal,
因为是非管理员账户,所以要下载update2的,否则无法自定义目录
wget https://developer.download.nvidia.com/compute/cuda/10.1/Prod/local_installers/cuda_10.1.243_418.87.00_linux.run
运行文件:
sh cuda_10.1.243_418.87.00_linux.run
接受协议:
安装选项暂时只选Toolkit:
然后进入Options进行,更改安装路径:
改为事先建立好的CUDA目录:
然后Done,回到Option选项下选择Library install path的设置:
更改为事先创建好的路径:
然后Done-Install进行安装。
安装完成,会看到如下提示,将路径添加到用户的环境变量中:
vim ~/.bash_profile
更改如下两行:
export PATH="/data/wuha