以下是在Anaconda中安装PyTorch 2.0并在Jupyter Notebook中使用它的步骤:
-
在Anaconda中创建一个新的虚拟环境。打开Anaconda Prompt或终端,并运行以下命令:
conda create --name pytorch2.0 python=3.8
这将创建一个名为"pytorch2.0"的新虚拟环境,使用Python 3.8版本。如果需要,您可以选择不同的Python版本。
-
激活虚拟环境:
conda activate pytorch2.0
-
使用适合您系统和硬件的命令安装PyTorch 2.0。请访问PyTorch官方网站,查找适用于您环境的PyTorch 2.0安装命令。例如,在Linux上安装支持Python 3.8的PyTorch 2.0与CUDA支持,您可以运行:
conda install pytorch=2.0 cudatoolkit=<version> -c pytorch
将
<version>
替换为您系统上安装的CUDA版本。如果您没有GPU或CUDA,可以省略cudatoolkit
参数以安装仅支持CPU的PyTorch 2.0版本。 -
在虚拟环境中安装Jupyter Notebook:
conda install jupyter
-
启动Jupyter Notebook:
jupyter notebook
-
在Jupyter Notebook界