2.dir与help函数
dir函数查看一个大类里面有什么小类
123456import torchprint(dir(torch.cuda))# return['Any', 'BFloat16Storage', 'BFloat16Tensor', 'BoolStorage', 'BoolTensor', 'ByteStorage', 'ByteTensor', 'CUDAGraph', 'CharStorage', 'CharTensor', 'ComplexDoubleStorage', 'ComplexFloatStorage', 'CudaError', 'DeferredCudaCallError', 'Device', 'Dict', 'DoubleStorage ...
1.准备工作
查看显卡信息1nvidia-smi
创建虚拟环境12345conda create -n Pytorch python=3.8conda activate Pytorchpip list # 查看当前虚拟环境中的包conda env list # 查看虚拟环境有哪些where python # 查看python解释器在哪里
将jupyter与创建的虚拟环境关联
在base环境下安装Pytorch(不推荐)
在创建好的虚拟环境中安装jupyter
123conda activate pytorchpip install nb_conda jupyter notebook
查看pytorch是否可用12import torchprint(torch.cuda.is_available())


