安装pytorch-GPU版

一、安查看NVIDIA版本

右键桌面:

安装pytorch-GPU版_第1张图片

 点击最下面系统信息:安装pytorch-GPU版_第2张图片

 查看组件:

安装pytorch-GPU版_第3张图片

 显卡驱动版本与cuda版本的对于关系图:

安装pytorch-GPU版_第4张图片

二、安装pytorch 

打开anaconda prompt,创建新的环境,命名为pytorch:

conda create -n pytorch python=3.8

 创建成功:

安装pytorch-GPU版_第5张图片

 切换到pytorch环境:

conda activate pytorch

换源:

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/win-64/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/win-64/
conda config --set show_channel_urls yes

安装pytorch 10.2版本:

conda install pytorch torchvision torchaudio cudatoolkit=10.2

下载:

安装pytorch-GPU版_第6张图片

 等待下载完成,发现测试还是false,由于下载很多较低版本都没有成功,破罐子破摔下载了11.3版本,发现成功了:

安装pytorch-GPU版_第7张图片

 

你可能感兴趣的:(pytorch,深度学习,python)