1.anaconda的下载和安装
2.cuda安装
判断是不是有独立显卡(计算机-我的电脑-管理-设备管理器-显示适配器)
确定显卡支持的cuda版本,(nvidia控制面板-帮助-系统信息-组件)
下载cuda对应的版本,官网下载- 选择cuda的安装环境
安装cuda,安装过程选择自定义-取消documentation下visual studio integration,记住安装路劲做path配置
配置cuda环境变量
验证cuda是否安装成功
打开cmd终端-输入nvcc -V
3.安装cuDNN
- 官网下载必须先注册
- 解压之后将所有的文件拷贝到cuda文件夹下面
4.安装pytorch
打开anaconda的prompt终端命令
打开pytorch官网选择指定的版本,复制安装命令在anaconda中选择并下载,后面就是继续等待,这里有梯子可能会很快
安装结束
测试是否安装成功,我在安装过程中出现如下错误,需要单独下载Microsoft Visual C++来进行安装下载地址,安装完成之后重启即可
1
2
3import torch
print(torch.__version__)
print(torch.cuda.is_available())
后面可以拿模型过来测试一下看看gpu使用加速时间
本文作者:
SparkParis
本文链接: https://sparkparis.github.io/2020/09/13/linux-pytorchconfig/
版权声明: 本作品采用 知识共享署名-非商业性使用-相同方式共享 4.0 国际许可协议 进行许可。转载请注明出处!
本文链接: https://sparkparis.github.io/2020/09/13/linux-pytorchconfig/
版权声明: 本作品采用 知识共享署名-非商业性使用-相同方式共享 4.0 国际许可协议 进行许可。转载请注明出处!