以windows11 为例安装conda,配置cuda和pytorch环境

最新推荐文章于 2024-03-20 09:29:49 发布

小哥不在

最新推荐文章于 2024-03-20 09:29:49 发布

阅读量1k

点赞数

文章标签： python conda ai

本文链接：https://blog.csdn.net/gaochuanlove/article/details/132604342

版权

以windows11 为例安装conda,配置cuda和pytorch环境

环境配置

anaconda环境安装

安装anaconda,从官网获取安装文件: https://www.anaconda.com/

安装后配置环境变量: 
ANACONDA_HOME=E:\ProgramData\anaconda3
path增加%ANACONDA_HOME% %ANACONDA_HOME%\Scripts %ANACONDA_HOME%\Library\bin %ANACONDA_HOME%\Library\mingw-w64\bin 几个执行目录

配置conda配置和包存储路径:
pkgs_dirs:
  - E:\compile\.conda\pkgs
envs_dirs
  - E:\compile\.conda\envs

anaconda环境配置常用操作

conda环境

# 创建一个python 3.11.4 的环境
conda create -n py311 python=3.11.4

# 安装好后，使用activate激活某个环境
activate py311 # for Windows
source activate py311 # for Linux & Mac
# 激活后，会发现terminal输入的地方多了(py311)的字样
(py311) PS E:\codespace\pure_python> python --version
Python 3.11.4
# 从conda环境解除
conda deactivate py311 # for Windows
source deactivate py311 # for Linux & Mac

# 删除一个已有的环境
conda remove --name py311 --all

#  查看当前conda环境
(py311) PS E:\codespace\pure_python> conda info -e 
# conda environments:
#
py311                 *  C:\Users\gaoch\.conda\envs\py311
base                     E:\ProgramData\anaconda3

验证conda环境

# 验证conda是否正常工作,测试安装numpy
conda install numpy
# conda会从从远程搜索numpy的相关信息和依赖项目

# 查看已经安装的packages
conda list
# 最新版的conda是从site-packages文件夹中搜索已经安装的包，可以显示出通过各种方式安装的包
# 查看某个指定环境的已安装包
conda list -n py311

# 查找package信息
conda search numpy
Loading channels: done
# Name                  Version           Build  Channel             
numpy                     1.5.1          py26_1  pkgs/free   

# 安装package
conda install -n py311 numpy
# 如果不用-n指定环境名称，则被安装在当前活跃环境
# 也可以通过-c指定通过某个channel安装

# 更新package
conda update -n py311 numpy
# 删除package
conda remove -n py311 numpy

# 更新conda，保持conda最新
conda update conda
# 更新anaconda
conda update anaconda
## 更新python
conda update python

解决下载速度慢得问题

## 查看当前conda配置,配置清华源
conda config --show channels
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud//pytorch/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/

## 设置搜索是显示通道地址
conda config --set show_channel_urls yes

##  设置 pip 清华源
pip config list
python -m pip install --upgrade pip
pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple

安装NVIDIA cuda计算平台

安装NVIDIA cuda计算平台
C:\Users\gaoch>nvidia-smi
Thu Jul 20 13:50:58 2023
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 522.06       Driver Version: 522.06       CUDA Version: 11.8     |
|-------------------------------+----------------------+----------------------+
| GPU  Name            TCC/WDDM | Bus-Id        Disp.A | Volatile Uncorr. ECC |
| Fan  Temp  Perf  Pwr:Usage/Cap|         Memory-Usage | GPU-Util  Compute M. |
|                               |                      |               MIG M. |
|===============================+======================+======================|
|   0  NVIDIA GeForce ... WDDM  | 00000000:01:00.0  On |                  N/A |
| N/A   43C    P8    10W /  N/A |    532MiB /  6144MiB |      5%      Default |
|                               |                      |                  N/A |
+-------------------------------+----------------------+----------------------+

+-----------------------------------------------------------------------------+
| Processes:                                                                  |
|  GPU   GI   CI        PID   Type   Process name                  GPU Memory |
|        ID   ID                                                   Usage      |
|=============================================================================|
|    0   N/A  N/A      3132    C+G   ...8bbwe\WindowsTerminal.exe    N/A      |
|    0   N/A  N/A      4824    C+G   ...y\ShellExperienceHost.exe    N/A      |
|    0   N/A  N/A      6176    C+G   ...cw5n1h2txyewy\LockApp.exe    N/A      |
|    0   N/A  N/A      9308    C+G   C:\Windows\explorer.exe         N/A      |
|    0   N/A  N/A     13520    C+G   ...n1h2txyewy\SearchHost.exe    N/A      |
|    0   N/A  N/A     13552    C+G   ...ge\Application\msedge.exe    N/A      |
|    0   N/A  N/A     13604    C+G   ...artMenuExperienceHost.exe    N/A      |
|    0   N/A  N/A     14664    C+G   ...2txyewy\TextInputHost.exe    N/A      |
|    0   N/A  N/A     16812    C+G   ...8wekyb3d8bbwe\GameBar.exe    N/A      |
|    0   N/A  N/A     17308    C+G   ...werToys.PowerLauncher.exe    N/A      |
|    0   N/A  N/A     19328    C+G   ...e\PhoneExperienceHost.exe    N/A      |
+-----------------------------------------------------------------------------+
官网获取cuda安装文件https://www.nvidia.com/zh-tw/geforce/technologies/cuda/,NVCC获取安装后cuda版本
C:\Users\gaoch>nvcc -V
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2022 NVIDIA Corporation
Built on Wed_Sep_21_10:41:10_Pacific_Daylight_Time_2022
Cuda compilation tools, release 11.8, V11.8.89
Build cuda_11.8.r11.8/compiler.31833905_0

获取当前安装的cuda版本为 CUDA Version: 11.8  ,从pytorch官网获取到可以使用的cuda版本
这里的cuda118 对用pytorch为2.0.1 可以使用pip安装或者conda安装
命令如下:
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia