赞
踩
1.电脑联想拯救者y7000,显卡GTX1650。
2.ubuntu版本18.04。
1.需要提前安装显卡驱动,具体安装过程在这里不加赘述,可以通过命令
nvidia-smi
查看英伟达显卡驱动版本。
2.安装cuda,通过nvidia-smi查看显卡驱动对应的cuda版本。
cuda是nvidia推出的用于自家GPU的并行计算框架,也就是说cuda只能在nvidia的GPU上运行,而且只有当要解决的计算问题是可以大量并行计算的时候才能发挥cuda的作用。
具体步骤参考链接:cuda安装教程
tip:在编写指令时注意cuda的版本。
3.安装Anaconda,注意下载和cuda对应的版本。
具体步骤参考链接:Anaconda安装教程
4.安装cudnn(此步可省略)
cudnn(CUDA Deep Neural Network library):是NVIDIA打造的针对深度神经网络的加速库,是一个用于深层神经网络的GPU加速库。如果你要用GPU训练模型,cudnn不是必须的,但是一般会采用这个加速库。
具体步骤参考链接:cudnn安装教程
5.配置pytorch虚拟环境
(1)配置环境变量供多用户使用Anaconda
(2)配置pytorch虚拟环境
注意:生成虚拟环境的路径应该为:
~/anaconda3/envs
1.查看显卡驱动版本
nvidia-smi
2.切换到pytorch虚拟环境下:
conda activate 虚拟环境(如pytorch)
3.下载依赖requirement.txt下载缓慢的解决方法
https://blog.csdn.net/hhxxtszj/article/details/128775991
pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple/
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。