Anaconda与pytorch安装配置详解

下载Anaconda

Anaconda | The World's Most Popular Data Science PlatformAnaconda is the birthplace of Python data science. We are a movement of data scientists, data-driven enterprises, and open source communities.https://www.anaconda.com/

然后自己配置环境变量:

接下来,进入NVIDIA控制面板,点击帮助-系统信息:

 可以查看到驱动程序版本。

这里安装 CUDA 11.6 。

安装CUDA

 下载安装包后启动,

 

 

 

 

测试,终端输入 nvcc -V ,返回CUDA版本号,即安装成功。

安装cuDNN

将压缩包解压,然后将其中的三个文件夹加入到cuda的安装目录下,

这里是 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.6

直接粘贴即可。

***现在的系统环境变量:

 在系统变量中:

安装

pytorch

PyTorchAn open source machine learning framework that accelerates the path from research prototyping to production deployment.https://pytorch.org/

 根据对应的型号版本来选择:

更多的型号版本选择:

 Previous PyTorch Versions | PyTorchAn open source machine learning framework that accelerates the path from research prototyping to production deployment.https://pytorch.org/get-started/previous-versions/

在里面找到对应的版本安装命令:

# CUDA 11.6
conda install pytorch==1.12.0 torchvision==0.13.0 torchaudio==0.12.0 cudatoolkit=11.6 -c pytorch -c conda-forge

注意,在输入上面的安装指令前, 还是先创建一个新环境:

conda create -n name python==3.10.9

现在是创建conda环境名为pytorch:

 

 

 输入 conda info --envs 查看现有环境

 

切换到创建的这个环境,

然后输入安装pytorch命令:

 

 

-------------------------------------------------------------------------------------------------------------------------------

 这里也给出了conda更新方法:

==> WARNING: A newer version of conda exists. <==
  current version: 23.1.0
  latest version: 23.3.1

Please update conda by running

    $ conda update -n base -c defaults conda

Or to minimize the number of packages updated during conda update use

     conda install conda=23.3.1

使用 conda update -n base -c defaults conda 就行。

-------------------------------------------------------------------------------------------------------------------------------

 

 

测试

最后进行演示:

至此,pytorch安装完成。

猜你喜欢

转载自blog.csdn.net/m0_72572822/article/details/129938788