win10 安装pytorch10.2 gpu成功

这篇博客详细记录了如何在已安装Anaconda3的基础上,创建一个名为'pytorch38'的环境,然后安装CUDA 10.2和对应的NVIDIA显卡驱动。接着,通过conda命令安装PyTorch 1.0.2、torchvision和torchaudio。最后,验证PyTorch安装成功的方法是通过运行`torch.cuda.is_available()`检查GPU是否可用。

安装pytorch10.2:

先安装anaconda3。我之前安过所以没下载。需要可以下载安装。

conda的命令行中使用conda create -n pytorch38 python=3.8建立环境。

nvidia-smi检查发现没有安装显卡驱动。

nvidia-smi检查是否安装400以上版本的显卡驱动。

官网上找到nvidia合适的显卡版本进行安装。

nvidia的显卡驱动,我安的是511.79.默认安装就可以。

完毕后安装pytorch

conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch

安装完毕,可能会比较慢。上G的包。我自己安装还好。

安装完毕测试是否可用。

import torch

torch.cuda.is_available()

在CentOS系统中安装GPU版本的PyTorch,主要步骤包含安装CUDA驱动、创建Anaconda虚拟环境以及安装PyTorch。以下是详细教程: ### 安装CUDA驱动 可以使用wget命令下载CUDA安装文件,然后使用sudo命令运行安装脚本。以CUDA 10.2为例,命令如下: ```bash wget https://developer.download.nvidia.com/compute/cuda/10.2/Prod/local_installers/cuda_10.2.89_440.33.01_linux.run sudo sh cuda_10.2.89_440.33.01_linux.run ``` 安装过程中,按照提示完成配置。 [^2] ### 创建Anaconda虚拟环境 如果系统中没有安装Anaconda,可以先从官网下载并安装安装完成后,创建一个新的虚拟环境: ```bash conda create -n pytorch_gpu python=3.9 conda activate pytorch_gpu ``` ### 安装GPU版本的PyTorch 根据CUDA版本,可以选择合适的PyTorch版本进行安装。如果已经将包下载到本地,例如包路径为`D:\pytorch`,可以使用pip命令进行安装: ```bash pip install D:\\pytorch\\torch-2.0.0+cu118-cp39-cp39-win_amd64.whl pip install D:\\pytorch\\torchvision-0.15.0+cu118-cp39-cp39-win_amd64.whl pip install D:\\pytorch\\torchaudio-2.0.0+cu118-cp39-cp39-win_amd64.whl ``` 若从网络安装,可根据PyTorch官网提供的命令进行安装,要确保选择与CUDA版本匹配的PyTorch版本。 [^1] ### 验证安装 安装完成后,可以在Python环境中验证PyTorch是否安装成功,并且GPU是否可用: ```python import torch print(torch.cuda.is_available()) ``` 如果输出为`True`,则表示GPU版本的PyTorch安装成功GPU可用。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值