【超级简单】3步 安装conda + pytorch gpu版本
- [1. 创建虚拟环境](#1. 创建虚拟环境)
-
- [至于python 3.X和cuda对应 ?](#至于python 3.X和cuda对应 ?)
- [2. 下载cuda 驱动,cuda toolkit (可选)](#2. 下载cuda 驱动,cuda toolkit (可选))
- [3. 虚拟环境中输入以下指令](#3. 虚拟环境中输入以下指令)
-
- [至于 怎么查看自己电脑对应的cuda版本号 ?](#至于 怎么查看自己电脑对应的cuda版本号 ?)
- 4.测试是否安装成功
下载anaconda,能让你高效管理不同版本运行环境。
conda + pytorch gpu
适用于linux 和window版本,
建议前提确保自己的显卡能用哈。
以下是在window10的示范。
1. 创建虚拟环境
conda create -n your_env_name python=3.X
conda avtivate your_env_name
至于python 3.X和cuda对应 ?
参考Tensorflow 的资料:
https://www.tensorflow.org/install/source_windows?hl=zh-cn
2. 下载cuda 驱动,cuda toolkit (可选)
https://developer.nvidia.com/cuda-toolkit-archive
下载成功的话可以 在cmd 输入 nvcc -V, 显示版本号。
显示 not found 则说明没转成功。
3. 虚拟环境中输入以下指令
conda install pytorch torchvision torchaudio cudatoolkit=X.X -c pytorch
X.X替换成你的版本号
至于 怎么查看自己电脑对应的cuda版本号 ?
:鼠标右键------>NVDIA控制面板------>帮助------>系统信息------>组件,product name 那一列就是nvdia对应版本。
4.测试是否安装成功
bash
torch.cuda.is_avaliable()
# True, gpu is ok
# false, try to fix it by rebooting up.
if not, there are some pro. with your settings, like conda environment doesnt correct, cuda Ver. is not right.
有疑惑可以再看看这篇文章写的。
笔墨多一点。
具体可参考:https://blog.csdn.net/qq_36851515/article/details/107608967