pytorch gpu版安装——亲测有效

一、查看cuda版本

在命令行中输入nvidia-smi

nvidia-smi

得到cuda版本为12.1,安装的cuda版本不高于12.1即可。

二、安装torch、torchvision、torchaudio三个组件

python3.9为例,当然其他版本也适用。

经验:

1. 安装cuda11.5(官网显示cu115)版本对应的三个组件,是比较稳妥的

2. 国内源容易在安装时自动替换为cpu版本,因此从pytorch官网下载较稳妥

3. 建议使用pip安装,conda安装很可能会安装为cpu版本

下面为具体步骤:

1)打开网址:https://download.pytorch.org/whl/cu102

首先选择torchctrl + F 搜索 [cu115-cp39-cp39-win] 这里cu115 是我们下载的 CUDA 11.5 版本,cp39-cp39 是说我们的 Python 版本是 3.9。如果要安装python其他版本修改为对应的版本即可。

单击即可下载。这里torch版本为1.11.0,我们要去官网查找该版本对应的torchvision torchaudio版本。ctrl + F 搜索 [pip install torch==1.11.0] 并且对应cuda11.5

在之前的网址https://download.pytorch.org/whl/cu102中选择torchaudioctrl + F 搜索 [cu115-cp39-cp39-win],选择版本为0.11.0的。高亮处单击下载。

同理在之前的网址中选择torchvisionctrl + F 搜索 [cu115-cp39-cp39-win],选择版本为0.11.0的。高亮处单击下载。

下载了3.whl文件,建议都安装到同一个文件夹下:

2)安装虚拟环境

默认大家都安装好Anaconda了。在开始菜单中搜索anaconda Prompt,点击进入。

创建python虚拟环境:

conda create -n torch115_gpu python=3.9

这里torch115_gpu表示你即将安装的虚拟环境的名字,表示python3.9版本。于是输入 conda create -n torch115_gpu python=3.9 后回车:

注意torch115_gpu文件夹可以在Anaconda的安装目录下的envs文件夹下找到。

之后激活该虚拟环境

conda activate torch115_gpu

那么这里就输入conda activate torch115_gpu 后回车

回车后的结果应该前面有一个(torch115_gpu),代表你现在已经进入了gym_gpu这个虚拟环境。

3)加载下载好的三个.whl文件并测试安装是否成功

现在需要依次将之前下载的三个.whl文件加载,需要这三个文件的绝对路径,依次按下面顺序输入(就是输入完一个就按一下回车等待安装完后再进行安装下一个)

pip install D:\pytorch_whl\torch-1.11.0+cu115-cp39-cp39-win_amd64.whl

pip install D:\pytorch_whl\torchaudio-0.11.0+cu115-cp39-cp39-win_amd64.whl

pip install D:\pytorch_whl\torchvision-0.12.0+cu115-cp39-cp39-win_amd64.whl

安装过程耐心等待。

安装结束后需要测试是否成功安装gpu版本的pytorch,这里依旧在torch115_gpu环境下输入python,进入python编程环境后输入import torch 回车后输入torch.cuda.is_available()。如果返回True则安装成功。

4)打开pycharm并打开需要配置环境的项目,会在右上角提示配置环境,点开配置环境选择conda>使用已有的环境就可以选择激活的torch111_gpu环境!

猜你喜欢

转载自blog.csdn.net/lzdjlu/article/details/134772588