vlambda博客
学习文章列表

1. 安装Anaconda，可以查看已经安装的环境
   conda env list

   为Pytorch创建新的环境，
   conda create -n pytorch python=3.8

2.  进入conda的Pytorch环境
   conda activate pytorch

conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch

 
   
   
 

  
    
    
  
 
   
   
 
 
   
   
 

  
    
    
  
 
   
 
   
 
   
 
   
 
  
  
    
    
  
pythonimport torchx = torch.rand(5, 3)print(x)
 
   
   
 
 
   
   
 

  
    
    
  安装成功的话，会得到下面的随机张量，
 
   
   
 

 
   
   
 

  
    
    
  tensor([[0.3380, 0.3845, 0.3217],
 [0.8337, 0.9050, 0.2650],
 [0.2979, 0.7141, 0.9069],
 [0.1449, 0.1132, 0.1375],
 [0.4675, 0.3947, 0.1426]])
 
   
   
 

6. 测试CUDA是否成功

 
   
   
 

  
    
    
  import torch
torch.cuda.is_available()