本文主要讲述了在Windows10系统中,通过anaconda安装GPU版本pytorch的具体步骤,即需要在Windows10电脑上已经安装好anaconda,anaconda的安装步骤可以参考此篇博客:
Anaconda安装步骤(Windows10)
本文主要分为三大部分
一、查看cuda版本
1.按下win+R键,输入cmd并按回车键,输入下面的命令行查看cuda版本
nvidia-smi
可以看到显卡驱动版本是496.49,最高支持的cuda版本是11.5,我们根据cuda的最高支持版本进行pytorch的安装

二、安装pytorch
1.打开anaconda prompt

2.输入下面的命令行,创建一个名称为pytorch的python3.7环境(名称和python版本任意)
conda create -n pytorch python=3.7

输入y

3.输入下面的命令,进入刚刚建好的名称为pytorch的环境中去
conda activate pytorch

4.输入下面的命令行,给pytorch环境换清华源
conda config --add channels https://mirrors.tuna

本文详细指导如何在Windows10系统中,通过已安装的Anaconda环境安装CUDA 11.3版本的PyTorch,包括检查CUDA版本、创建新环境、更换源、安装步骤及PyCharm导入验证。
最低0.47元/天 解锁文章
2540





