学校实验室和pytorch不兼容怎么办

Question

学校服务器跟电脑的pytorch不兼容怎么办
我是用vscode远程连的服务器
本来我的cuda是10.2，不兼容之后搜网上的教程我就把pytorch和cuda都卸载了
我是在自己Windows系统上重新安cuda11.6吗
但是刚刚尝试装了几遍一直安装失败
是不是应该在服务器上安装cuda11.6，那是在linux系统上安装吗
有没有人帮帮我😭

Answer 1

学校服务器操作系统一般是 Linux，而自己的电脑是windows系统，这两个是完全不一样的，所以服务器cuda版本的是11.2，和自己电脑的cuda版本是没有影响的，明白这一点，以确保正确地安装 conda 和 PyTorch。

1 下载conda

链接：Anaconda官网
在终端运行脚本命令

bash Anaconda3-2023.09-0-Linux-x86_64.sh

2 查看cuda版本

nvidia-smi

3 创建自己的环境

使用conda创建一个名为 pytorch的虚拟环境，并指定Python版本为3.8。

conda create -n pytorch python=3.8

进入我们刚刚安装好的环境，我们要在pytorch这个环境下安装，不要在base环境下安装

conda activate pytorch

4 下载pytorch

在pytorch官网查找符合自己cuda版本的pytorch
网页往下滑，点击previous versions of pytorch
需要安装的是cuda版本为12.0 的pytorch，找了半天并没有CUDA12.0的版本，可以选择的是CUDA 11.6
CUDA版本通常是向下兼容的。你可以在安装了CUDA 11.6的系统上使用适用于CUDA 12.0的PyTorch

conda install pytorch==1.13.1 torchvision==0.14.1 torchaudio==0.13.1 pytorch-cuda=11.6 -c pytorch -c nvidia

在终端输入，等待运行

5 验证pytorch是否安装成功

python

import torch
torch.cuda.is_available()

出现True表示安装成功