1. 安装Anaconda
去Anaconda官方网址下载,选择python3版本。
2. 安装Cuda
去Cuda官网下载相应程序包,根据程序包的格式进行安装。 注意和显卡的兼容性。
安装后配置环境变量:在/etc/profile文件的最后添加
export PATH=/usr/local/cuda-version/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-version/lib64:$LD_LIBRARY_PATH
使profile文件生效后,在命令行中输入nvcc -V,若能看到相关信息,即说明安装成功。若未成功,请参考Cuda安装踩坑指南。
3. 安装Pytorch
去Pytorch官网,选择相应的系统和软件版本,按提示的命令进行安装。
4. 安装matplotlib
$conda install matplotlib
5. 安装OpenCV
$conda install opencv
6. 安装tqdm
$conda install tqdm
7. 安装fastai
$conda install -c fastai fastai
8. 安装配置jupyter notebook
$conda install jupyter notebook
远程配置详情参考。